KaizeShi commited on
Commit
4cbc990
·
1 Parent(s): 72dd3d9

Add application file

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -56,7 +56,8 @@ else:
56
  BASE_MODEL,
57
  device_map={"": device},
58
  low_cpu_mem_usage=True,
59
- load_in_8bit=True,
 
60
  )
61
  model = PeftModel.from_pretrained(
62
  model,
 
56
  BASE_MODEL,
57
  device_map={"": device},
58
  low_cpu_mem_usage=True,
59
+ load_in_8bit=False,
60
+ torch_dtype=torch.float16,
61
  )
62
  model = PeftModel.from_pretrained(
63
  model,