Spaces:
Sleeping
Sleeping
Commit ·
306f49b
1
Parent(s): 9b1d484
Updated
Browse files
main.py
CHANGED
|
@@ -22,7 +22,7 @@ from transformers import pipeline
|
|
| 22 |
# CONFIGURATION
|
| 23 |
# ===============================================
|
| 24 |
API_SECRET = "techdisciplesai404"
|
| 25 |
-
MODEL_NAME = "meta-llama/Llama-
|
| 26 |
DEVICE = 0 if torch.cuda.is_available() else -1
|
| 27 |
|
| 28 |
# ===============================================
|
|
|
|
| 22 |
# CONFIGURATION
|
| 23 |
# ===============================================
|
| 24 |
API_SECRET = "techdisciplesai404"
|
| 25 |
+
MODEL_NAME = "meta-llama/Llama-3.1-8B"
|
| 26 |
DEVICE = 0 if torch.cuda.is_available() else -1
|
| 27 |
|
| 28 |
# ===============================================
|