Spaces:
				
			
			
	
			
			
		Build error
		
	
	
	
			
			
	
	
	
	
		
		
		Build error
		
	Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | @@ -195,8 +195,8 @@ def load_and_prepare_model(): | |
| 195 |  | 
| 196 | 
             
                #pipe.unet.to(memory_format=torch.channels_last)
         | 
| 197 | 
             
                #pipe.enable_vae_tiling()
         | 
| 198 | 
            -
                pipe.unet = torch.compile(pipe.unet, backend="hidet")
         | 
| 199 | 
            -
                 | 
| 200 | 
             
                pipe.to(device=device, dtype=torch.bfloat16)
         | 
| 201 |  | 
| 202 | 
             
                return pipe
         | 
|  | |
| 195 |  | 
| 196 | 
             
                #pipe.unet.to(memory_format=torch.channels_last)
         | 
| 197 | 
             
                #pipe.enable_vae_tiling()
         | 
| 198 | 
            +
                #pipe.unet = torch.compile(pipe.unet, backend="hidet")
         | 
| 199 | 
            +
                pipe.unet = torch.compile(pipe.unet, backend="cudagraphs", dynamic=False)
         | 
| 200 | 
             
                pipe.to(device=device, dtype=torch.bfloat16)
         | 
| 201 |  | 
| 202 | 
             
                return pipe
         |