Spaces:
Runtime error
Runtime error
Commit
·
a56a74d
1
Parent(s):
8ed0878
Update app_modules/utils.py
Browse files- app_modules/utils.py +1 -2
app_modules/utils.py
CHANGED
|
@@ -352,13 +352,12 @@ def load_tokenizer_and_model(base_model,adapter_model,load_8bit=False):
|
|
| 352 |
)
|
| 353 |
else:
|
| 354 |
model = LlamaForCausalLM.from_pretrained(
|
| 355 |
-
base_model, device_map={"": device}, low_cpu_mem_usage=True
|
| 356 |
)
|
| 357 |
model = PeftModel.from_pretrained(
|
| 358 |
model,
|
| 359 |
adapter_model,
|
| 360 |
device_map={"": device},
|
| 361 |
-
torch_dtype=torch.float16
|
| 362 |
)
|
| 363 |
|
| 364 |
if not load_8bit:
|
|
|
|
| 352 |
)
|
| 353 |
else:
|
| 354 |
model = LlamaForCausalLM.from_pretrained(
|
| 355 |
+
base_model, device_map={"": device}, low_cpu_mem_usage=True
|
| 356 |
)
|
| 357 |
model = PeftModel.from_pretrained(
|
| 358 |
model,
|
| 359 |
adapter_model,
|
| 360 |
device_map={"": device},
|
|
|
|
| 361 |
)
|
| 362 |
|
| 363 |
if not load_8bit:
|