Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | 
         @@ -123,7 +123,6 @@ def load_and_prepare_model(model_id): 
     | 
|
| 123 | 
         
             
                #sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear")
         
     | 
| 124 | 
         
             
                #pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0").to(torch.bfloat16)
         
     | 
| 125 | 
         
             
                #pipeX = StableDiffusionXLPipeline.from_pretrained("ford442/Juggernaut-XI-v11-fp32",use_safetensors=True)
         
     | 
| 126 | 
         
            -
                
         
     | 
| 127 | 
         
             
                pipe = StableDiffusionXLPipeline.from_pretrained(
         
     | 
| 128 | 
         
             
                    'ford442/RealVisXL_V5.0_BF16',
         
     | 
| 129 | 
         
             
                   #'ford442/Juggernaut-XI-v11-fp32',
         
     | 
| 
         @@ -149,7 +148,7 @@ def load_and_prepare_model(model_id): 
     | 
|
| 149 | 
         
             
                #pipe.scheduler = DPMSolverMultistepScheduler.from_pretrained('SG161222/RealVisXL_V5.0', subfolder='scheduler', algorithm_type='sde-dpmsolver++')
         
     | 
| 150 | 
         
             
                #pipe.vae = vaeX
         
     | 
| 151 | 
         
             
                #pipe.unet = unetX
         
     | 
| 152 | 
         
            -
                 
     | 
| 153 | 
         
             
                #pipe.scheduler = sched
         
     | 
| 154 | 
         
             
                #pipe.vae=vae.to(torch.bfloat16)
         
     | 
| 155 | 
         
             
                #pipe.unet=pipeX.unet
         
     | 
| 
         @@ -228,7 +227,7 @@ def uploadNote(): 
     | 
|
| 228 | 
         
             
                    f.write(f"Model VAE: juggernaut to bfloat before cuda then attn_proc \n")
         
     | 
| 229 | 
         
             
                    f.write(f"Model UNET: default ford442/RealVisXL_V5.0_BF16 \n")
         
     | 
| 230 | 
         
             
                    f.write(f"Model HiDiffusion OFF \n")
         
     | 
| 231 | 
         
            -
                    f.write(f"Model do_resize  
     | 
| 232 | 
         
             
                upload_to_ftp(filename) 
         
     | 
| 233 | 
         | 
| 234 | 
         
             
            @spaces.GPU(duration=30)
         
     | 
| 
         | 
|
| 123 | 
         
             
                #sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear")
         
     | 
| 124 | 
         
             
                #pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0").to(torch.bfloat16)
         
     | 
| 125 | 
         
             
                #pipeX = StableDiffusionXLPipeline.from_pretrained("ford442/Juggernaut-XI-v11-fp32",use_safetensors=True)
         
     | 
| 
         | 
|
| 126 | 
         
             
                pipe = StableDiffusionXLPipeline.from_pretrained(
         
     | 
| 127 | 
         
             
                    'ford442/RealVisXL_V5.0_BF16',
         
     | 
| 128 | 
         
             
                   #'ford442/Juggernaut-XI-v11-fp32',
         
     | 
| 
         | 
|
| 148 | 
         
             
                #pipe.scheduler = DPMSolverMultistepScheduler.from_pretrained('SG161222/RealVisXL_V5.0', subfolder='scheduler', algorithm_type='sde-dpmsolver++')
         
     | 
| 149 | 
         
             
                #pipe.vae = vaeX
         
     | 
| 150 | 
         
             
                #pipe.unet = unetX
         
     | 
| 151 | 
         
            +
                pipe.vae.do_resize=False
         
     | 
| 152 | 
         
             
                #pipe.scheduler = sched
         
     | 
| 153 | 
         
             
                #pipe.vae=vae.to(torch.bfloat16)
         
     | 
| 154 | 
         
             
                #pipe.unet=pipeX.unet
         
     | 
| 
         | 
|
| 227 | 
         
             
                    f.write(f"Model VAE: juggernaut to bfloat before cuda then attn_proc \n")
         
     | 
| 228 | 
         
             
                    f.write(f"Model UNET: default ford442/RealVisXL_V5.0_BF16 \n")
         
     | 
| 229 | 
         
             
                    f.write(f"Model HiDiffusion OFF \n")
         
     | 
| 230 | 
         
            +
                    f.write(f"Model do_resize OFF \n")
         
     | 
| 231 | 
         
             
                upload_to_ftp(filename) 
         
     | 
| 232 | 
         | 
| 233 | 
         
             
            @spaces.GPU(duration=30)
         
     |