Spaces:
Runtime error
Runtime error
Commit ·
f4e8f75
1
Parent(s): e1ec073
commit 000424
Browse files
app.py
CHANGED
|
@@ -35,7 +35,7 @@ swagger = Swagger(app, template=swagger_template, config=swagger_config)
|
|
| 35 |
login(token=os.environ.get("CHAT_MATE"))
|
| 36 |
|
| 37 |
# ✅ Load LLaMA 3 Instruct model
|
| 38 |
-
model_id = "meta-llama/
|
| 39 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 40 |
model = AutoModelForCausalLM.from_pretrained(
|
| 41 |
model_id, torch_dtype="auto", device_map="auto"
|
|
|
|
| 35 |
login(token=os.environ.get("CHAT_MATE"))
|
| 36 |
|
| 37 |
# ✅ Load LLaMA 3 Instruct model
|
| 38 |
+
model_id = "meta-llama/Llama-3.1-8B-Instruct"
|
| 39 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 40 |
model = AutoModelForCausalLM.from_pretrained(
|
| 41 |
model_id, torch_dtype="auto", device_map="auto"
|