Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,8 @@ model_name = "NLPnorth/snakmodel-7b-instruct"
|
|
| 31 |
model = AutoModelForCausalLM.from_pretrained(
|
| 32 |
model_name,
|
| 33 |
torch_dtype="auto",
|
| 34 |
-
device_map="auto"
|
|
|
|
| 35 |
)
|
| 36 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 37 |
|
|
|
|
| 31 |
model = AutoModelForCausalLM.from_pretrained(
|
| 32 |
model_name,
|
| 33 |
torch_dtype="auto",
|
| 34 |
+
device_map="auto",
|
| 35 |
+
low_cpu_mem_usage=True,
|
| 36 |
)
|
| 37 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 38 |
|