Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | @@ -22,7 +22,7 @@ if not torch.cuda.is_available(): | |
| 22 | 
             
            MAX_SEED = np.iinfo(np.int32).max
         | 
| 23 | 
             
            CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "1") == "1"
         | 
| 24 | 
             
            MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4192"))
         | 
| 25 | 
            -
            USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", " | 
| 26 | 
             
            ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
         | 
| 27 |  | 
| 28 | 
             
            style_list = [
         | 
| @@ -79,8 +79,6 @@ pipe = PixArtAlphaPipeline.from_pretrained( | |
| 79 | 
             
                    torch_dtype=torch.float16,
         | 
| 80 | 
             
                    use_safetensors=True,
         | 
| 81 | 
             
            ).to("cuda:0")
         | 
| 82 | 
            -
            quantize(pipe.transformer, weights=qfloat8)
         | 
| 83 | 
            -
            freeze(pipe.transformer)
         | 
| 84 |  | 
| 85 | 
             
            def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
         | 
| 86 | 
             
                p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
         | 
|  | |
| 22 | 
             
            MAX_SEED = np.iinfo(np.int32).max
         | 
| 23 | 
             
            CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "1") == "1"
         | 
| 24 | 
             
            MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4192"))
         | 
| 25 | 
            +
            USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "1") == "1"
         | 
| 26 | 
             
            ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
         | 
| 27 |  | 
| 28 | 
             
            style_list = [
         | 
|  | |
| 79 | 
             
                    torch_dtype=torch.float16,
         | 
| 80 | 
             
                    use_safetensors=True,
         | 
| 81 | 
             
            ).to("cuda:0")
         | 
|  | |
|  | |
| 82 |  | 
| 83 | 
             
            def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
         | 
| 84 | 
             
                p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
         | 
 
			
