Revert "Changed to 1B"
Browse filesThis reverts commit b7e1b4b8d6430aa710878a7e7fec12ffcc127942.
app.py
CHANGED
|
@@ -13,7 +13,7 @@ from datetime import datetime
|
|
| 13 |
|
| 14 |
class LlamaChat:
|
| 15 |
def __init__(self):
|
| 16 |
-
self.model_name = "meta-llama/Llama-3.2-
|
| 17 |
self.device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 18 |
self.tokenizer = None
|
| 19 |
self.model = None
|
|
|
|
| 13 |
|
| 14 |
class LlamaChat:
|
| 15 |
def __init__(self):
|
| 16 |
+
self.model_name = "meta-llama/Llama-3.2-3B-Instruct"
|
| 17 |
self.device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 18 |
self.tokenizer = None
|
| 19 |
self.model = None
|