davidizzle commited on
Commit
7eae56b
·
1 Parent(s): 697b3ed

Do not trust remote code

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ model = AutoModelForCausalLM.from_pretrained(model_id,
14
  # torch_dtype=torch.float32,
15
  device_map="auto",
16
  torch_dtype=torch.float16,
17
- trust_remote_code=True
18
  )
19
  # model.to("cpu")
20
 
 
14
  # torch_dtype=torch.float32,
15
  device_map="auto",
16
  torch_dtype=torch.float16,
17
+ trust_remote_code=False
18
  )
19
  # model.to("cpu")
20