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
|