Skip to content

fix: DDIM clamp opt-in, negative prompt wiring, EMA bug, latent_dir, memory_format flag#2

Merged
atandra2000 merged 2 commits into
mainfrom
fix/inference-bugs
Jun 2, 2026
Merged

fix: DDIM clamp opt-in, negative prompt wiring, EMA bug, latent_dir, memory_format flag#2
atandra2000 merged 2 commits into
mainfrom
fix/inference-bugs

Conversation

@atandra2000
Copy link
Copy Markdown
Owner

Five bug fixes from the engineering audit.

…efault, memory_format flag

- DDIMScheduler: make pred_x0 clamp opt-in via clamp_pred_x0 param (default False)
- inference.py: wire --negative prompt through to uncond embedding
- SD_ImageGen.py: pass model.unet to ema.apply_shadow()
- SD_Train.py/train.py/SD_Train_v2.py: fix latent_dir default (remove double nesting)
- SD_Train.py/train.py: add --memory_format CLI flag to gate channels_last
  (use 'contiguous' for AMD/Apple Silicon)
@atandra2000 atandra2000 merged commit e86eaf9 into main Jun 2, 2026
1 check failed
@atandra2000 atandra2000 deleted the fix/inference-bugs branch June 2, 2026 08:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant