Spaces:
Build error
Build error
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
|