PengWeixuanSZU commited on
Commit
0d4e4ae
·
verified ·
1 Parent(s): 72202c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ random_seed = 42
56
  video_length = 201
57
  W = 1024
58
  H = W
59
- device = "cuda" if torch.cuda.is_available() else "cpu"
60
 
61
  def get_pipe_image_and_video_predictor():
62
  vae = AutoencoderKLWan.from_pretrained("./model/vae", torch_dtype=torch.float16)
 
56
  video_length = 201
57
  W = 1024
58
  H = W
59
+ device = "cpu" if torch.cuda.is_available() else "cpu"
60
 
61
  def get_pipe_image_and_video_predictor():
62
  vae = AutoencoderKLWan.from_pretrained("./model/vae", torch_dtype=torch.float16)