Spaces:
Running
Running
daydayup1225
commited on
Commit
·
2613d29
1
Parent(s):
0b7aaec
Update app_modules/utils.py
Browse files- app_modules/utils.py +2 -2
app_modules/utils.py
CHANGED
@@ -446,8 +446,8 @@ def load_finetune_tokenizer_and_model(base_model_name_or_path, load_8bit=False):
|
|
446 |
base_model_name_or_path, device_map={"": device}, low_cpu_mem_usage=True
|
447 |
)
|
448 |
|
449 |
-
if not load_8bit:
|
450 |
-
|
451 |
|
452 |
model.eval()
|
453 |
return tokenizer, model, device
|
|
|
446 |
base_model_name_or_path, device_map={"": device}, low_cpu_mem_usage=True
|
447 |
)
|
448 |
|
449 |
+
# if not load_8bit:
|
450 |
+
# model.half() # seems to fix bugs for some users.
|
451 |
|
452 |
model.eval()
|
453 |
return tokenizer, model, device
|