asimcodeml commited on
Commit
95717ac
·
verified ·
1 Parent(s): c20b056

add load 4bit=true

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -75,7 +75,8 @@ model = AutoModelForCausalLM.from_pretrained(
75
  MODEL_PATH,
76
  trust_remote_code=True,
77
  device_map="auto", # Use GPU if available
78
- low_cpu_mem_usage=True
 
79
  )
80
 
81
  pipe = pipeline(
 
75
  MODEL_PATH,
76
  trust_remote_code=True,
77
  device_map="auto", # Use GPU if available
78
+ low_cpu_mem_usage=True,
79
+ load_in_4bit=True
80
  )
81
 
82
  pipe = pipeline(