1inkusFace commited on
Commit
b3dd2ff
·
verified ·
1 Parent(s): 3e16887

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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"To cuda and bfloat \n")
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)