SkyReels_B

Paused

1inkusFace commited on Mar 9

Commit

f8c19dd

verified ·

1 Parent(s): bb1eaa6

Update skyreelsinfer/skyreels_video_infer.py

Files changed (1) hide show

skyreelsinfer/skyreels_video_infer.py CHANGED Viewed

@@ -95,7 +95,7 @@ class SkyReelsVideoInfer:
         from .offload import Offload
         self.pipe = self._load_model( #No : SkyreelsVideoPipeline
-            model_id=self.model_id, quant_model=self.quant_model, device="cpu"
         )
         if self.is_offload and self.offload_config:

         from .offload import Offload
         self.pipe = self._load_model( #No : SkyreelsVideoPipeline
+            model_id=self.model_id, quant_model=self.quant_model, device="cuda"
         )
         if self.is_offload and self.offload_config: