Spaces:
Runtime error
Runtime error
mohan1869
commited on
Commit
·
b973723
1
Parent(s):
cd3d513
Deploy SQLCoder with Streamlit
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ def load_model():
|
|
12 |
model = AutoModelForCausalLM.from_pretrained(
|
13 |
model_name,
|
14 |
device_map="auto",
|
15 |
-
|
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
|