Spaces:
				
			
			
	
			
			
		Sleeping
		
	
	
	
			
			
	
	
	
	
		
		
		Sleeping
		
	Update src/TinyLLama/text_generation.py
Browse files
    	
        src/TinyLLama/text_generation.py
    CHANGED
    
    | 
         @@ -1,6 +1,6 @@ 
     | 
|
| 1 | 
         
             
            from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
         
     | 
| 2 | 
         
             
            from src.classmodels.inputforgeneration import InputForGeneration
         
     | 
| 3 | 
         
            -
            from errorlog.errorlog import log_error
         
     | 
| 4 | 
         
             
            from pathlib import Path
         
     | 
| 5 | 
         | 
| 6 | 
         
             
            #MODEL NAME AS PER IN HUGGING FACE :- TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
         
     | 
| 
         | 
|
| 1 | 
         
             
            from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
         
     | 
| 2 | 
         
             
            from src.classmodels.inputforgeneration import InputForGeneration
         
     | 
| 3 | 
         
            +
            from src.errorlog.errorlog import log_error
         
     | 
| 4 | 
         
             
            from pathlib import Path
         
     | 
| 5 | 
         | 
| 6 | 
         
             
            #MODEL NAME AS PER IN HUGGING FACE :- TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
         
     |