GrayShine commited on
Commit
ee503e4
1 Parent(s): 9985965

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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()