ford442 commited on
Commit
e24b5ca
·
verified ·
1 Parent(s): 2f1c53d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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
- #pipe.vae.do_resize=False
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 ON \n")
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)