Spaces:

anntnikita
/

matrix-game-demo

Runtime error

anntnikita commited on Aug 18

Commit

56e5104

verified ·

1 Parent(s): 842ef5c

Fix torch_dtype assignment in pipeline loading

Files changed (1) hide show

app.py CHANGED Viewed

@@ -96,8 +96,7 @@ class MatrixGame:
             try:
                 pipeline = AutoPipelineForVideo.from_pretrained(
                     self.MODEL_ID,
-                    torch_dtype=
                     use_auth_token=self.hf_token,
                 )
             except Exception as e:
@@ -108,7 +107,6 @@ class MatrixGame:
                 pipeline = ImageToVideoPipeline.from_pretrained(
                     self.MODEL_ID,
                     torch_dtype=self.dtype,
                     use_auth_token=self.hf_token,
                 )
             except Exception as e:

             try:
                 pipeline = AutoPipelineForVideo.from_pretrained(
                     self.MODEL_ID,
+                    torch_dtype=self.dtype,
                     use_auth_token=self.hf_token,
                 )
             except Exception as e:
                 pipeline = ImageToVideoPipeline.from_pretrained(
                     self.MODEL_ID,
                     torch_dtype=self.dtype,
                     use_auth_token=self.hf_token,
                 )
             except Exception as e: