Spaces:
Runtime error
Runtime error
upd
Browse files
app.py
CHANGED
@@ -71,7 +71,7 @@ def load_lora(lora_path, base_model="decapoda-research/llama-7b-hf"):
|
|
71 |
# load_in_8bit=True,
|
72 |
# device_map=device_map,
|
73 |
low_cpu_mem_usage=True,
|
74 |
-
torch_type=torch.float16,
|
75 |
)
|
76 |
print("Loading LoRA...")
|
77 |
lora = PeftModel.from_pretrained(
|
|
|
71 |
# load_in_8bit=True,
|
72 |
# device_map=device_map,
|
73 |
low_cpu_mem_usage=True,
|
74 |
+
# torch_type=torch.float16,
|
75 |
)
|
76 |
print("Loading LoRA...")
|
77 |
lora = PeftModel.from_pretrained(
|