Fix lint

xorbitsai · Aug 8, 2024 · 6180cb6 · 6180cb6
1 parent 26d63c4
commit 6180cb6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/xinference/model/video/core.py b/xinference/model/video/core.py
@@ -164,6 +164,7 @@ def create_video_model_instance(
     model_spec = match_diffusion(model_name, download_hub)
     if not model_path:
         model_path = cache(model_spec)
+    assert model_path is not None
 
     model = DiffUsersVideoModel(
         model_uid,

diff --git a/xinference/model/video/diffusers.py b/xinference/model/video/diffusers.py
@@ -94,7 +94,7 @@ def text_to_video(
             "diffusers text_to_video args: %s",
             kwargs,
         )
-        # assert callable(self._model)
+        assert self._model is not None
         prompt_embeds, _ = self._model.encode_prompt(
             prompt=prompt,
             do_classifier_free_guidance=True,
@@ -103,6 +103,7 @@ def text_to_video(
             device=self._model.device,
             dtype=torch.float16,
         )
+        assert callable(self._model)
         output = self._model(
             num_inference_steps=50,
             guidance_scale=6,