harmionestark commited on
Commit
a35806e
Β·
verified Β·
1 Parent(s): dcc1c7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -19,7 +19,6 @@ pipe.to(device)
19
  pipe.text_encoder.to(device, dtype=torch.float32)
20
  pipe.vae.to(device, dtype=torch.float32)
21
  pipe.unet.to(device, dtype=torch.float32)
22
- pipe.scheduler.to(device, dtype=torch.float32)
23
 
24
  def force_float32(model):
25
  for param in model.parameters():
@@ -30,7 +29,6 @@ def force_float32(model):
30
  force_float32(pipe.text_encoder)
31
  force_float32(pipe.vae)
32
  force_float32(pipe.unet)
33
- force_float32(pipe.scheduler)
34
 
35
  MAX_SEED = np.iinfo(np.int32).max
36
  MAX_IMAGE_SIZE = 1024
 
19
  pipe.text_encoder.to(device, dtype=torch.float32)
20
  pipe.vae.to(device, dtype=torch.float32)
21
  pipe.unet.to(device, dtype=torch.float32)
 
22
 
23
  def force_float32(model):
24
  for param in model.parameters():
 
29
  force_float32(pipe.text_encoder)
30
  force_float32(pipe.vae)
31
  force_float32(pipe.unet)
 
32
 
33
  MAX_SEED = np.iinfo(np.int32).max
34
  MAX_IMAGE_SIZE = 1024