abhisheksan commited on
Commit
32aea62
1 Parent(s): 6feef58

Update model name in ModelManager to use Llama-1B-Instruct for improved performance

Browse files
Files changed (1) hide show
  1. app/services/poetry_generation.py +1 -1
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-2B-Instruct"
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)