VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-06 13:47:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

doc: add reference link for input vector gradient requirement in Chroma class

Browse Source
This commit is contained in:
Kohya S
2025-07-28 22:08:02 +09:00
parent af14eab6d7
commit 6c8973c2da
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
library/chroma_models.py
View File

@@ -695,6 +695,7 @@ class Chroma(Flux):
input_vec = self.get_input_vec(timesteps, guidance, img.shape[0])
# kohya-ss: I'm not sure why requires_grad is set to True here
# original code: https://github.com/lodestone-rock/flow/blob/c76f63058980d0488826936025889e256a2e0458/src/models/chroma/model.py#L217
input_vec.requires_grad = True
mod_vectors = self.distilled_guidance_layer(input_vec)
else: