Test

Paused

Carlos s commited on Oct 3, 2025

Commit

88fe1bb

verified ·

1 Parent(s): 5802895

Update video_service.py

Files changed (1) hide show

video_service.py CHANGED Viewed

@@ -259,6 +259,7 @@ class VideoService:
             local_dir=os.getenv("HF_HOME"),
             cache_dir=os.getenv("HF_HOME_CACHE"),
             token=os.getenv("HF_TOKEN"),
         )
         self.config["checkpoint_path"] = distilled_model_path

             local_dir=os.getenv("HF_HOME"),
             cache_dir=os.getenv("HF_HOME_CACHE"),
             token=os.getenv("HF_TOKEN"),
+            torch_dtype="Float8_e4m3fn",
         )
         self.config["checkpoint_path"] = distilled_model_path