XiangpengYang commited on
Commit
8008f96
·
1 Parent(s): 91db0c0
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -274,6 +274,8 @@ class VideoCoF_Controller(Wan_Controller):
274
  # Ensure generator is created on the same device as the pipeline's transformer
275
  gen_device = getattr(getattr(self, "pipeline", None), "transformer", None)
276
  gen_device = gen_device.device if gen_device is not None else self.device
 
 
277
  generator = torch.Generator(device=gen_device).manual_seed(int(seed_textbox))
278
 
279
  try:
 
274
  # Ensure generator is created on the same device as the pipeline's transformer
275
  gen_device = getattr(getattr(self, "pipeline", None), "transformer", None)
276
  gen_device = gen_device.device if gen_device is not None else self.device
277
+ if gen_device.type == 'meta':
278
+ gen_device = self.device
279
  generator = torch.Generator(device=gen_device).manual_seed(int(seed_textbox))
280
 
281
  try: