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*"
|