Spaces:
Runtime error
Runtime error
shivam9980
commited on
Commit
•
cf6bb41
1
Parent(s):
b79b0fa
Update app.py
Browse files
app.py
CHANGED
@@ -4,9 +4,9 @@ from unsloth import FastLanguageModel
|
|
4 |
import torch
|
5 |
model, tokenizer = FastLanguageModel.from_pretrained(
|
6 |
model_name = "shivam9980/mistral-7b-news-cnn-merged", # Choose ANY! eg teknium/OpenHermes-2.5-Mistral-7B
|
7 |
-
max_seq_length =
|
8 |
-
dtype =
|
9 |
-
load_in_4bit =
|
10 |
token = hf_token, # use one if using gated models like meta-llama/Llama-2-7b-hf
|
11 |
)
|
12 |
|
|
|
4 |
import torch
|
5 |
model, tokenizer = FastLanguageModel.from_pretrained(
|
6 |
model_name = "shivam9980/mistral-7b-news-cnn-merged", # Choose ANY! eg teknium/OpenHermes-2.5-Mistral-7B
|
7 |
+
max_seq_length = 2048,
|
8 |
+
dtype = None,
|
9 |
+
load_in_4bit = True,
|
10 |
token = hf_token, # use one if using gated models like meta-llama/Llama-2-7b-hf
|
11 |
)
|
12 |
|