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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -239,7 +239,7 @@ def video_generation(text, image, scfg_scale, tcfg_scale, img_cfg_scale, diffusi
239
  vae.to(dtype=torch.float16)
240
  model.to(dtype=torch.float16)
241
  text_encoder.to(dtype=torch.float16)
242
- image_encoder.to(dtype=torch.float16)
243
  use_fp16 = True
244
  print('Initialization Finished')
245
 
@@ -296,7 +296,7 @@ def video_prediction(text, image, scfg_scale, tcfg_scale, img_cfg_scale, prefram
296
  vae.to(dtype=torch.float16)
297
  model.to(dtype=torch.float16)
298
  text_encoder.to(dtype=torch.float16)
299
- image_encoder.to(dtype=torch.float16)
300
  use_fp16 = True
301
  print('Initialization Finished')
302
 
 
239
  vae.to(dtype=torch.float16)
240
  model.to(dtype=torch.float16)
241
  text_encoder.to(dtype=torch.float16)
242
+ # image_encoder.to(dtype=torch.float16)
243
  use_fp16 = True
244
  print('Initialization Finished')
245
 
 
296
  vae.to(dtype=torch.float16)
297
  model.to(dtype=torch.float16)
298
  text_encoder.to(dtype=torch.float16)
299
+ # image_encoder.to(dtype=torch.float16)
300
  use_fp16 = True
301
  print('Initialization Finished')
302