MiniMax-Remover

Running on Zero

PengWeixuanSZU commited on Jun 20

Commit

72202c9

verified ·

1 Parent(s): 6e69110

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -56,7 +56,7 @@ random_seed = 42
 video_length = 201
 W = 1024
 H = W
-device = "cuda" #if torch.cuda.is_available() else "cpu"
 def get_pipe_image_and_video_predictor():
     vae = AutoencoderKLWan.from_pretrained("./model/vae", torch_dtype=torch.float16)

 video_length = 201
 W = 1024
 H = W
+device = "cuda" if torch.cuda.is_available() else "cpu"
 def get_pipe_image_and_video_predictor():
     vae = AutoencoderKLWan.from_pretrained("./model/vae", torch_dtype=torch.float16)