Spaces:
Running
on
Zero
Running
on
Zero
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)
|