Spaces:

facat
/

alpaca-lora-cn

Runtime error

facat commited on Mar 28, 2023

Commit

a44a7b9

•

1 Parent(s): b290f0c

upd

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,7 +68,7 @@ def evaluate(
 def load_lora(lora_path, base_model="decapoda-research/llama-7b-hf"):
     model = LlamaForCausalLM.from_pretrained(
         base_model,
-        load_in_8bit=True,
         # device_map=device_map,
         low_cpu_mem_usage=True,
         torch_type=torch.float16,

 def load_lora(lora_path, base_model="decapoda-research/llama-7b-hf"):
     model = LlamaForCausalLM.from_pretrained(
         base_model,
+        # load_in_8bit=True,
         # device_map=device_map,
         low_cpu_mem_usage=True,
         torch_type=torch.float16,