Med Tiouti commited on
Commit
0e1fedf
1 Parent(s): c22681b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ model = AutoModelForCausalLM.from_pretrained(
32
  load_in_4bit=True,
33
  torch_dtype=torch.float16,
34
  low_cpu_mem_usage=True,
35
- trust_remote_code=True
36
  cache_dir=cache_path
37
  )
38
  max_len = 8192
 
32
  load_in_4bit=True,
33
  torch_dtype=torch.float16,
34
  low_cpu_mem_usage=True,
35
+ trust_remote_code=True,
36
  cache_dir=cache_path
37
  )
38
  max_len = 8192