Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ def load_model(hf_token):
|
|
| 11 |
model_name=model_name,
|
| 12 |
load_in_4bit=True,
|
| 13 |
token=hf_token,
|
| 14 |
-
use_zero=True, # Ensure ZeroGPU usage
|
| 15 |
)
|
| 16 |
return model, tokenizer
|
| 17 |
except Exception as e:
|
|
|
|
| 11 |
model_name=model_name,
|
| 12 |
load_in_4bit=True,
|
| 13 |
token=hf_token,
|
| 14 |
+
# use_zero=True, # Ensure ZeroGPU usage
|
| 15 |
)
|
| 16 |
return model, tokenizer
|
| 17 |
except Exception as e:
|