Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -118,7 +118,7 @@ def load_and_prepare_model():
|
|
118 |
token=HF_TOKEN,
|
119 |
text_encoder=None,
|
120 |
text_encoder_2=None,
|
121 |
-
|
122 |
)
|
123 |
|
124 |
'''
|
@@ -135,13 +135,13 @@ def load_and_prepare_model():
|
|
135 |
`force_upcast` can be set to `False` - see: https://huggingface.co/madebyollin/sdxl-vae-fp16-fix
|
136 |
|
137 |
'''
|
138 |
-
|
139 |
pipe.to(device=device, dtype=torch.bfloat16)
|
140 |
-
pipe.vae.to(device=device, dtype=torch.bfloat16)
|
141 |
-
pipe.vae.do_resize=False
|
142 |
#pipe.vae.do_rescale=False
|
143 |
#pipe.vae.do_convert_rgb=True
|
144 |
-
pipe.vae.vae_scale_factor=8 #pipe.unet.set_default_attn_processor()
|
145 |
pipe.vae.set_default_attn_processor()
|
146 |
print(f'Pipeline: ')
|
147 |
#print(f'_optional_components: {pipe._optional_components}')
|
|
|
118 |
token=HF_TOKEN,
|
119 |
text_encoder=None,
|
120 |
text_encoder_2=None,
|
121 |
+
vae=None,
|
122 |
)
|
123 |
|
124 |
'''
|
|
|
135 |
`force_upcast` can be set to `False` - see: https://huggingface.co/madebyollin/sdxl-vae-fp16-fix
|
136 |
|
137 |
'''
|
138 |
+
pipe.vae=vaeX
|
139 |
pipe.to(device=device, dtype=torch.bfloat16)
|
140 |
+
#pipe.vae.to(device=device, dtype=torch.bfloat16)
|
141 |
+
#pipe.vae.do_resize=False
|
142 |
#pipe.vae.do_rescale=False
|
143 |
#pipe.vae.do_convert_rgb=True
|
144 |
+
#pipe.vae.vae_scale_factor=8 #pipe.unet.set_default_attn_processor()
|
145 |
pipe.vae.set_default_attn_processor()
|
146 |
print(f'Pipeline: ')
|
147 |
#print(f'_optional_components: {pipe._optional_components}')
|