nodes/IamCreateAI-Ruyi-Models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix autocast device type

Browse Source
This commit is contained in:
cellzero
2025-01-20 20:21:28 +08:00
parent f6543017c9
commit 6c7b5972dc
3 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
predict_i2v.py
+1 -1
View File
@@ -270,7 +270,7 @@ pipeline.transformer.tea_cache.initialize(tea_cache_enabled, tea_cache_threshold
pipeline.transformer.enhance_a_video.initialize(enhance_a_video_enabled, enhance_a_video_weight, enhance_a_video_skip_start_steps, enhance_a_video_skip_end_steps, steps)
# Generate video
with torch.no_grad(), torch.autocast(str(device), dtype = pipeline.transformer.dtype):
with torch.no_grad(), torch.autocast(device_type=device.type, dtype=pipeline.transformer.dtype):
video_length = int(video_length // pipeline.vae.mini_batch_encoder * pipeline.vae.mini_batch_encoder) if video_length != 1 else 1
input_video, input_video_mask, clip_image = get_image_to_video_latent(start_img, end_img, video_length=video_length, sample_size=(height, width))