theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 10:18:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix rand device

Browse Source
This commit is contained in:
Artiprocher
2025-01-15 19:30:38 +08:00
parent 9f8112ec34
commit 75d62a02d1
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
diffsynth/pipelines/hunyuan_video.py
View File

@@ -89,6 +89,7 @@ class HunyuanVideoPipeline(BasePipeline):
input_video=None, input_video=None,
denoising_strength=1.0, denoising_strength=1.0,
seed=None, seed=None,
rand_device=None,
height=720, height=720,
width=1280, width=1280,
num_frames=129, num_frames=129,
@@ -109,7 +110,8 @@ class HunyuanVideoPipeline(BasePipeline):
self.scheduler.set_timesteps(num_inference_steps, denoising_strength) self.scheduler.set_timesteps(num_inference_steps, denoising_strength)
# Initialize noise # Initialize noise
noise = self.generate_noise((1, 16, (num_frames - 1) // 4 + 1, height//8, width//8), seed=seed, device=self.device, dtype=self.torch_dtype) rand_device = self.device if rand_device is None else rand_device
noise = self.generate_noise((1, 16, (num_frames - 1) // 4 + 1, height//8, width//8), seed=seed, device=rand_device, dtype=self.torch_dtype).to(self.device)
if input_video is not None: if input_video is not None:
self.load_models_to_device(['vae_encoder']) self.load_models_to_device(['vae_encoder'])
input_video = self.preprocess_images(input_video) input_video = self.preprocess_images(input_video)