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

Deleting meaningless

Browse Source
This commit is contained in:
Stan Travis
2025-01-03 20:21:38 +03:00
committed by GitHub
parent 963f2c9bf4
commit e2a64fbe35
1 changed files with 0 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfyui/comfyui_nodes.py
-5
View File
@@ -424,11 +424,6 @@ class Ruyi_I2VSampler:
# Load control embeddings # Load control embeddings
embeddings = self.get_control_embeddings(pipeline, aspect_ratio, motion, camera_direction) embeddings = self.get_control_embeddings(pipeline, aspect_ratio, motion, camera_direction)
try:
torch.cuda.reset_peak_memory_stats(device)
except:
pass
# Inference # Inference
with torch.no_grad(), torch.autocast(str(device), dtype = pipeline.transformer.dtype): with torch.no_grad(), torch.autocast(str(device), 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 video_length = int(video_length // pipeline.vae.mini_batch_encoder * pipeline.vae.mini_batch_encoder) if video_length != 1 else 1