vilarin commited on
Commit
a13aa7f
·
verified ·
1 Parent(s): 57ab448

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -43,6 +43,7 @@ model = AutoModelForCausalLM.from_pretrained(
43
  MODEL,
44
  torch_dtype=torch.bfloat16,
45
  device_map="auto",
 
46
  quantization_config=quantization_config)
47
 
48
  @spaces.GPU(duration=100)
 
43
  MODEL,
44
  torch_dtype=torch.bfloat16,
45
  device_map="auto",
46
+ trust_remote_code=True,
47
  quantization_config=quantization_config)
48
 
49
  @spaces.GPU(duration=100)