Wenye He commited on
Commit
9d099a3
·
verified ·
1 Parent(s): dd8d3db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ class ChatModel:
41
  quantization_config=bnb_config,
42
  device_map="auto",
43
  torch_dtype=torch.float16,
44
- trust_remote_code=True
45
  )
46
 
47
  self.models[model_name] = model
 
41
  quantization_config=bnb_config,
42
  device_map="auto",
43
  torch_dtype=torch.float16,
44
+ trust_remote_code=False
45
  )
46
 
47
  self.models[model_name] = model