Spaces:
Sleeping
Sleeping
Commit
·
552a4ed
1
Parent(s):
1d1ee9e
updated app.py
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
|
|
| 37 |
if HF_TOKEN is None:
|
| 38 |
raise ValueError("HF_TOKEN is not set. Check your .env file.")
|
| 39 |
login(HF_TOKEN, add_to_git_credential=True)
|
| 40 |
-
MODEL_NAME = "mistralai/Mistral-7B-
|
| 41 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=HF_TOKEN)
|
| 42 |
model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map="auto",use_auth_token=HF_TOKEN)
|
| 43 |
|
|
|
|
| 37 |
if HF_TOKEN is None:
|
| 38 |
raise ValueError("HF_TOKEN is not set. Check your .env file.")
|
| 39 |
login(HF_TOKEN, add_to_git_credential=True)
|
| 40 |
+
MODEL_NAME = "mistralai/Mistral-7B-4bit"
|
| 41 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=HF_TOKEN)
|
| 42 |
model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map="auto",use_auth_token=HF_TOKEN)
|
| 43 |
|