mohan1869 commited on
Commit
b973723
·
1 Parent(s): cd3d513

Deploy SQLCoder with Streamlit

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def load_model():
12
  model = AutoModelForCausalLM.from_pretrained(
13
  model_name,
14
  device_map="auto",
15
- load_in_8bit=True, # Use int8 quantization
16
  low_cpu_mem_usage=True,
17
  )
18
  return tokenizer, model
 
12
  model = AutoModelForCausalLM.from_pretrained(
13
  model_name,
14
  device_map="auto",
15
+ # load_in_8bit=True, # Use int8 quantization
16
  low_cpu_mem_usage=True,
17
  )
18
  return tokenizer, model