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- | 
| 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)
         |