update

projects/video_diffusion_sr/infer.py

@@ -41,7 +41,7 @@ from models.dit_v2 import na
 class VideoDiffusionInfer():
     def __init__(self, config: DictConfig):
         self.config = config
-        self.device = "cuda" if torch.cuda.is_available() else "cpu"
+        self.device = "cuda"
 
     def get_condition(self, latent: Tensor, latent_blur: Tensor, task: str) -> Tensor:
         t, h, w, c = latent.shape