Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	Commit 
							
							·
						
						d7f17ba
	
1
								Parent(s):
							
							52b1a93
								
Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | 
         @@ -22,7 +22,7 @@ login(token=os.environ.get('HF_KEY')) 
     | 
|
| 22 | 
         
             
            device = 'cpu'
         
     | 
| 23 | 
         
             
            seed = 10000
         
     | 
| 24 | 
         
             
            torch.cuda.max_memory_allocated(device='cpu')
         
     | 
| 25 | 
         
            -
            pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0",  
     | 
| 26 | 
         
             
            pipe = pipe.to(device)
         
     | 
| 27 | 
         
             
            # pipe.enable_xformers_memory_efficient_attention()
         
     | 
| 28 | 
         
             
            # torch.cuda.empty_cache()
         
     | 
| 
         | 
|
| 22 | 
         
             
            device = 'cpu'
         
     | 
| 23 | 
         
             
            seed = 10000
         
     | 
| 24 | 
         
             
            torch.cuda.max_memory_allocated(device='cpu')
         
     | 
| 25 | 
         
            +
            pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", use_safetensors=True)
         
     | 
| 26 | 
         
             
            pipe = pipe.to(device)
         
     | 
| 27 | 
         
             
            # pipe.enable_xformers_memory_efficient_attention()
         
     | 
| 28 | 
         
             
            # torch.cuda.empty_cache()
         
     |