(self,
config,
checkpoint_dir,
use_usp=False,
ulysses_size=None,
ring_size=None)
| 478 | class WanVaceMP(WanVace): |
| 479 | |
| 480 | def __init__(self, |
| 481 | config, |
| 482 | checkpoint_dir, |
| 483 | use_usp=False, |
| 484 | ulysses_size=None, |
| 485 | ring_size=None): |
| 486 | self.config = config |
| 487 | self.checkpoint_dir = checkpoint_dir |
| 488 | self.use_usp = use_usp |
| 489 | os.environ['MASTER_ADDR'] = 'localhost' |
| 490 | os.environ['MASTER_PORT'] = '12345' |
| 491 | os.environ['RANK'] = '0' |
| 492 | os.environ['WORLD_SIZE'] = '1' |
| 493 | self.in_q_list = None |
| 494 | self.out_q = None |
| 495 | self.inference_pids = None |
| 496 | self.ulysses_size = ulysses_size |
| 497 | self.ring_size = ring_size |
| 498 | self.dynamic_load() |
| 499 | |
| 500 | self.device = 'cpu' if torch.cuda.is_available() else 'cpu' |
| 501 | self.vid_proc = VaceVideoProcessor( |
| 502 | downsample=tuple( |
| 503 | [x * y for x, y in zip(config.vae_stride, config.patch_size)]), |
| 504 | min_area=480 * 832, |
| 505 | max_area=480 * 832, |
| 506 | min_fps=self.config.sample_fps, |
| 507 | max_fps=self.config.sample_fps, |
| 508 | zero_start=True, |
| 509 | seq_len=32760, |
| 510 | keep_last=True) |
| 511 | |
| 512 | def dynamic_load(self): |
| 513 | if hasattr(self, 'inference_pids') and self.inference_pids is not None: |
nothing calls this directly
no test coverage detected