anntnikita commited on
Commit
842ef5c
·
verified ·
1 Parent(s): 52aedc6

Remove variant parameter from pipeline loading to improve compatibility with latest diffusers.

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -96,8 +96,8 @@ class MatrixGame:
96
  try:
97
  pipeline = AutoPipelineForVideo.from_pretrained(
98
  self.MODEL_ID,
99
- torch_dtype=self.dtype,
100
- variant="fp16" if self.dtype == torch.float16 else None,
101
  use_auth_token=self.hf_token,
102
  )
103
  except Exception as e:
@@ -108,7 +108,7 @@ class MatrixGame:
108
  pipeline = ImageToVideoPipeline.from_pretrained(
109
  self.MODEL_ID,
110
  torch_dtype=self.dtype,
111
- variant="fp16" if self.dtype == torch.float16 else None,
112
  use_auth_token=self.hf_token,
113
  )
114
  except Exception as e:
 
96
  try:
97
  pipeline = AutoPipelineForVideo.from_pretrained(
98
  self.MODEL_ID,
99
+ torch_dtype=
100
+
101
  use_auth_token=self.hf_token,
102
  )
103
  except Exception as e:
 
108
  pipeline = ImageToVideoPipeline.from_pretrained(
109
  self.MODEL_ID,
110
  torch_dtype=self.dtype,
111
+
112
  use_auth_token=self.hf_token,
113
  )
114
  except Exception as e: