Spaces:
				
			
			
	
			
			
		Running
		
			on 
			
			Zero
	
	
	
			
			
	
	
	
	
		
		
		Running
		
			on 
			
			Zero
	Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | @@ -14,10 +14,7 @@ tokenizer = AutoTokenizer.from_pretrained(llasa_3b) | |
| 14 | 
             
            model = AutoModelForCausalLM.from_pretrained(
         | 
| 15 | 
             
                llasa_3b,
         | 
| 16 | 
             
                trust_remote_code=True,
         | 
| 17 | 
            -
                use_cache=False,
         | 
| 18 | 
            -
                torch_dtype=torch.bfloat16,
         | 
| 19 | 
             
                device_map='cuda',
         | 
| 20 | 
            -
                return_dict=True
         | 
| 21 | 
             
            )
         | 
| 22 |  | 
| 23 | 
             
            model_path = "srinivasbilla/xcodec2"
         | 
|  | |
| 14 | 
             
            model = AutoModelForCausalLM.from_pretrained(
         | 
| 15 | 
             
                llasa_3b,
         | 
| 16 | 
             
                trust_remote_code=True,
         | 
|  | |
|  | |
| 17 | 
             
                device_map='cuda',
         | 
|  | |
| 18 | 
             
            )
         | 
| 19 |  | 
| 20 | 
             
            model_path = "srinivasbilla/xcodec2"
         | 
