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

Get the correct frame with Chrono

Browse Source
This commit is contained in:
Disty0
2025-11-02 19:21:08 +03:00
parent fc4033a628
commit 5ba74b7263
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+7 -2
View File
@@ -16,6 +16,11 @@ debug = os.environ.get('SD_DIFFUSERS_DEBUG', None) is not None
last_p = None
orig_pipeline = shared.sd_model
image_frame_index = {
'WanPipeline': 0,
'ChronoEditPipeline': -1,
}
def restore_state(p: processing.StableDiffusionProcessing):
if p.state in ['reprocess_refine', 'reprocess_detail']:
@@ -197,7 +202,7 @@ def process_base(p: processing.StableDiffusionProcessing):
shared.log.debug(f'Generated: frames={output.frames[0].shape[1]}')
else:
shared.log.debug(f'Generated: frames={len(output.frames[0])}')
output.images = output.frames[0]
output.images = output.frames[image_frame_index.get(shared.sd_model.__class__.__name__, 0)]
if hasattr(output, 'images') and isinstance(output.images, np.ndarray):
output.images = torch.from_numpy(output.images)
except AssertionError as e:
@@ -437,7 +442,7 @@ def process_decode(p: processing.StableDiffusionProcessing, output):
if output is not None:
if not hasattr(output, 'images') and hasattr(output, 'frames'):
shared.log.debug(f'Generated: frames={len(output.frames[0])}')
output.images = output.frames[0]
output.images = output.frames[image_frame_index.get(shared.sd_model.__class__.__name__, 0)]
if output.images is not None and len(output.images) > 0 and isinstance(output.images[0], Image.Image):
return output.images
model = shared.sd_model if not is_refiner_enabled(p) else shared.sd_refiner