Spaces:
Configuration error
Configuration error
Update app.py
Browse files
app.py
CHANGED
|
@@ -205,7 +205,7 @@ def init_model():
|
|
| 205 |
vae.to(dtype=torch.float16)
|
| 206 |
model.to(dtype=torch.float16)
|
| 207 |
text_encoder.to(dtype=torch.float16)
|
| 208 |
-
image_encoder.to(dtype=torch.float16)
|
| 209 |
use_fp16 = True
|
| 210 |
print('Initialization Finished')
|
| 211 |
init_model()
|
|
|
|
| 205 |
vae.to(dtype=torch.float16)
|
| 206 |
model.to(dtype=torch.float16)
|
| 207 |
text_encoder.to(dtype=torch.float16)
|
| 208 |
+
# image_encoder.to(dtype=torch.float16)
|
| 209 |
use_fp16 = True
|
| 210 |
print('Initialization Finished')
|
| 211 |
init_model()
|