Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|