GrayShine commited on
Commit
ca7cb64
1 Parent(s): 653a139

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -234,7 +234,7 @@ def video_generation(text, image, scfg_scale, tcfg_scale, img_cfg_scale, diffusi
234
  print("xformer!")
235
  else:
236
  raise ValueError("xformers is not available. Make sure it is installed correctly")
237
- if args.use_fp16:
238
  print('Warnning: using half percision for inferencing!')
239
  vae.to(dtype=torch.float16)
240
  model.to(dtype=torch.float16)
@@ -291,7 +291,7 @@ def video_prediction(text, image, scfg_scale, tcfg_scale, img_cfg_scale, prefram
291
  print("xformer!")
292
  else:
293
  raise ValueError("xformers is not available. Make sure it is installed correctly")
294
- if args.use_fp16:
295
  print('Warnning: using half percision for inferencing!')
296
  vae.to(dtype=torch.float16)
297
  model.to(dtype=torch.float16)
 
234
  print("xformer!")
235
  else:
236
  raise ValueError("xformers is not available. Make sure it is installed correctly")
237
+ if use_fp16:
238
  print('Warnning: using half percision for inferencing!')
239
  vae.to(dtype=torch.float16)
240
  model.to(dtype=torch.float16)
 
291
  print("xformer!")
292
  else:
293
  raise ValueError("xformers is not available. Make sure it is installed correctly")
294
+ if use_fp16:
295
  print('Warnning: using half percision for inferencing!')
296
  vae.to(dtype=torch.float16)
297
  model.to(dtype=torch.float16)