Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
8008f96
1
Parent(s):
91db0c0
new
Browse files
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:
|