Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | @@ -21,8 +21,6 @@ def load_llama_model(): | |
| 21 | 
             
                    device_map="auto",
         | 
| 22 | 
             
                    # offload_folder="offload", 
         | 
| 23 | 
             
                ).to("cuda")
         | 
| 24 | 
            -
             | 
| 25 | 
            -
                )
         | 
| 26 | 
             
                processor = AutoProcessor.from_pretrained(model_id)
         | 
| 27 |  | 
| 28 | 
             
                return model, processor
         | 
|  | |
| 21 | 
             
                    device_map="auto",
         | 
| 22 | 
             
                    # offload_folder="offload", 
         | 
| 23 | 
             
                ).to("cuda")
         | 
|  | |
|  | |
| 24 | 
             
                processor = AutoProcessor.from_pretrained(model_id)
         | 
| 25 |  | 
| 26 | 
             
                return model, processor
         | 
