Spaces:
Sleeping
Sleeping
abhisheksan
commited on
Commit
•
32aea62
1
Parent(s):
6feef58
Update model name in ModelManager to use Llama-1B-Instruct for improved performance
Browse files
app/services/poetry_generation.py
CHANGED
@@ -25,7 +25,7 @@ class ModelManager:
|
|
25 |
|
26 |
def __init__(self):
|
27 |
if not ModelManager._initialized:
|
28 |
-
model_name = "meta-llama/Llama-
|
29 |
|
30 |
# Initialize tokenizer and model
|
31 |
self.tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
|
25 |
|
26 |
def __init__(self):
|
27 |
if not ModelManager._initialized:
|
28 |
+
model_name = "meta-llama/Llama-1B-Instruct"
|
29 |
|
30 |
# Initialize tokenizer and model
|
31 |
self.tokenizer = AutoTokenizer.from_pretrained(model_name)
|