Emmanuel Frimpong Asante commited on
Commit
8952525
·
1 Parent(s): 986aab5

update space

Browse files
backend/app/services/llama_service.py CHANGED
@@ -33,7 +33,7 @@ def load_llama_model():
33
  global llama_model, llama_tokenizer
34
  try:
35
  logger.info("Loading Llama 3.2 model and tokenizer.")
36
- model_name = "meta-llama/Llama-2-7b-hf" # Updated model name
37
 
38
  # Load tokenizer and model
39
  llama_tokenizer = AutoTokenizer.from_pretrained(model_name)
 
33
  global llama_model, llama_tokenizer
34
  try:
35
  logger.info("Loading Llama 3.2 model and tokenizer.")
36
+ model_name = "meta-llama/Llama-3.2-1B" # Updated model name
37
 
38
  # Load tokenizer and model
39
  llama_tokenizer = AutoTokenizer.from_pretrained(model_name)