Pass sample width/height/num_frames into cached sample prompt configs so reference sizing matches generation

This commit is contained in:
Jaret Burkett committed 2026-10-23 12:20:39 -06:00
1 parent 7861f92b13
commit 1beba8a96d
2 files changed
+4 -1

No files matched your search

@@ -170,8 +170,11 @@ class SDTrainer(BaseSDTrainProcess):
# needed so we can autoparse the prompt to handle flags
gen_img_config = GenerateImageConfig(
prompt=prompt, # it will autoparse the prompt
width=sample_item.width,
height=sample_item.height,
negative_prompt=sample_item.neg,
output_path=output_path,
num_frames=sample_item.num_frames,
ctrl_img=sample_item.ctrl_img,
ctrl_img_1=sample_item.ctrl_img_1,
ctrl_img_2=sample_item.ctrl_img_2,