facat commited on
Commit
97b8320
1 Parent(s): a44a7b9
Files changed (1) hide show
  1. app.py +1 -1
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(