Update factual/green_score/green.py
Browse files
factual/green_score/green.py
CHANGED
@@ -45,7 +45,7 @@ def tqdm_on_main(*args, **kwargs):
|
|
45 |
|
46 |
|
47 |
class GREEN:
|
48 |
-
def __init__(self, model_name, output_dir=".", cpu=
|
49 |
super().__init__()
|
50 |
warnings.filterwarnings(
|
51 |
"ignore", message="A decoder-only architecture is being used*"
|
|
|
45 |
|
46 |
|
47 |
class GREEN:
|
48 |
+
def __init__(self, model_name, output_dir=".", cpu=False):
|
49 |
super().__init__()
|
50 |
warnings.filterwarnings(
|
51 |
"ignore", message="A decoder-only architecture is being used*"
|