GGLS commited on
Commit
112adb1
·
verified ·
1 Parent(s): 7261d63

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -31,7 +31,6 @@ def load_model(model_name):
31
  f"{root_path}/{model_name}",
32
  device_map="auto",
33
  torch_dtype=torch.bfloat16,
34
- load_in_4bit=True,
35
  trust_remote_code=True,
36
  )
37
 
 
31
  f"{root_path}/{model_name}",
32
  device_map="auto",
33
  torch_dtype=torch.bfloat16,
 
34
  trust_remote_code=True,
35
  )
36