asimcodeml commited on
Commit
16ecba2
·
verified ·
1 Parent(s): fc19172

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ model = AutoModelForCausalLM.from_pretrained(
9
  MODEL_PATH,
10
  trust_remote_code=True,
11
  device_map="auto",
12
- low_cpu_mem_usage=True,
13
- quantization_config=None
14
  )
15
 
16
  pipe = pipeline(
 
9
  MODEL_PATH,
10
  trust_remote_code=True,
11
  device_map="auto",
12
+ low_cpu_mem_usage=True
13
+
14
  )
15
 
16
  pipe = pipeline(