Spaces:
Running
Running
daydayup1225
commited on
Commit
•
bd2a332
1
Parent(s):
7fe8d23
Update app_modules/utils.py
Browse files- app_modules/utils.py +2 -2
app_modules/utils.py
CHANGED
@@ -407,8 +407,8 @@ def load_tokenizer_and_model(base_model, adapter_model, load_8bit=False):
|
|
407 |
device_map={"": device},
|
408 |
)
|
409 |
|
410 |
-
if not load_8bit:
|
411 |
-
|
412 |
|
413 |
model.eval()
|
414 |
return tokenizer, model, device
|
|
|
407 |
device_map={"": device},
|
408 |
)
|
409 |
|
410 |
+
# if not load_8bit:
|
411 |
+
# model.half() # seems to fix bugs for some users.
|
412 |
|
413 |
model.eval()
|
414 |
return tokenizer, model, device
|