fix: pass max_size=800 to progress bar preview (was 85px wide)
The third element in ComfyUI's preview tuple is max_size in pixels, not JPEG quality. Passing 85 was capping the live loss curve at 85×40px. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -489,7 +489,7 @@ class SelvaLoraTrainer:
|
|||||||
# Live preview: send updated loss curve to ComfyUI frontend
|
# Live preview: send updated loss curve to ComfyUI frontend
|
||||||
preview_img = _draw_loss_curve(loss_history, log_interval, start_step)
|
preview_img = _draw_loss_curve(loss_history, log_interval, start_step)
|
||||||
pbar_train.update_absolute(
|
pbar_train.update_absolute(
|
||||||
step - start_step, remaining, ("JPEG", preview_img, 85)
|
step - start_step, remaining, ("JPEG", preview_img, 800)
|
||||||
)
|
)
|
||||||
|
|
||||||
if step % save_every == 0 or step == steps:
|
if step % save_every == 0 or step == steps:
|
||||||
|
|||||||
Reference in New Issue
Block a user