ford442 commited on
Commit
c0ef521
·
verified ·
1 Parent(s): dd17079

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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
- #vae=vaeX,
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}')