Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -42,7 +42,10 @@ model = AutoPeftModelForCausalLM.from_pretrained(
|
|
42 |
model_id, # YOUR MODEL YOU USED FOR TRAINING
|
43 |
load_in_4bit = load_in_4bit,
|
44 |
)
|
45 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
|
|
|
|
|
|
46 |
|
47 |
model.config.sliding_window = 4096
|
48 |
model.eval()
|
|
|
42 |
model_id, # YOUR MODEL YOU USED FOR TRAINING
|
43 |
load_in_4bit = load_in_4bit,
|
44 |
)
|
45 |
+
tokenizer = AutoTokenizer.from_pretrained(
|
46 |
+
model_id,
|
47 |
+
quantization_config=BitsAndBytesConfig(load_in_8bit=True)
|
48 |
+
)
|
49 |
|
50 |
model.config.sliding_window = 4096
|
51 |
model.eval()
|