Spaces:
Runtime error
Runtime error
project-baize
commited on
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:
|