Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-08-26 15:16:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #5033 from vladmandic/fix/video-frame-count

Browse Source 
fix(video): count generated frames on the first output axis
This commit is contained in:
Vladimir Mandic
2026-08-17 08:05:23 +02:00
committed by GitHub
parent 6b499b7d17 7792b1ab62
commit c053fb6c95
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+1 -4
View File
@@ -204,10 +204,7 @@ def process_base(p: processing.StableDiffusionProcessing):
t1 = time.time()
log.debug(f'Profile: pipeline call: {t1-t0:.2f}')
if not hasattr(output, 'images') and hasattr(output, 'frames'):
if hasattr(output.frames[0], 'shape'):
log.debug(f'Generated: frames={output.frames[0].shape[1]}')
else:
log.debug(f'Generated: frames={len(output.frames[0])}')
log.debug(f'Generated: frames={len(output.frames[0])}')
output.images = output.frames[0]
if hasattr(output, 'images') and isinstance(output.images, np.ndarray):
output.images = torch.from_numpy(output.images)