Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -110,6 +110,7 @@ def load_and_prepare_model():
|
|
110 |
#pipe.vae.vae_scale_factor=8
|
111 |
#pipe.to(device)
|
112 |
#pipe.to(torch.bfloat16)
|
|
|
113 |
print(f'init noise scale: {pipe.scheduler.init_noise_sigma}')
|
114 |
pipe.watermark=None
|
115 |
pipe.safety_checker=None
|
@@ -170,7 +171,7 @@ def uploadNote(prompt,num_inference_steps,guidance_scale,timestamp):
|
|
170 |
f.write(f"Steps: {num_inference_steps} \n")
|
171 |
f.write(f"Guidance Scale: {guidance_scale} \n")
|
172 |
f.write(f"SPACE SETUP: \n")
|
173 |
-
f.write(f"
|
174 |
upload_to_ftp(filename)
|
175 |
|
176 |
@spaces.GPU(duration=30)
|
|
|
110 |
#pipe.vae.vae_scale_factor=8
|
111 |
#pipe.to(device)
|
112 |
#pipe.to(torch.bfloat16)
|
113 |
+
pipe.unet.fuse_qkv_projections()
|
114 |
print(f'init noise scale: {pipe.scheduler.init_noise_sigma}')
|
115 |
pipe.watermark=None
|
116 |
pipe.safety_checker=None
|
|
|
171 |
f.write(f"Steps: {num_inference_steps} \n")
|
172 |
f.write(f"Guidance Scale: {guidance_scale} \n")
|
173 |
f.write(f"SPACE SETUP: \n")
|
174 |
+
f.write(f"fuse_qkv_projections \n")
|
175 |
upload_to_ftp(filename)
|
176 |
|
177 |
@spaces.GPU(duration=30)
|