Commit
·
ec5de02
1
Parent(s):
f7fa740
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ print(f"Starting to load the model {model_name}")
|
|
| 20 |
m = AutoModelForCausalLM.from_pretrained(
|
| 21 |
model_name,
|
| 22 |
device_map=0,
|
| 23 |
-
load_in_8bit=True,
|
| 24 |
)
|
| 25 |
|
| 26 |
m.config.pad_token_id = m.config.eos_token_id
|
|
|
|
| 20 |
m = AutoModelForCausalLM.from_pretrained(
|
| 21 |
model_name,
|
| 22 |
device_map=0,
|
| 23 |
+
#load_in_8bit=True,
|
| 24 |
)
|
| 25 |
|
| 26 |
m.config.pad_token_id = m.config.eos_token_id
|