support SD3 LoRA

2026-03-23 00:58:11 +00:00 · 2024-07-10 10:07:02 +08:00
parent 8113f95278
commit 979a8814f1
13 changed files with 1030 additions and 32 deletions
--- a/diffsynth/schedulers/flow_match.py
+++ b/diffsynth/schedulers/flow_match.py
@@ -40,3 +40,8 @@ class FlowMatchScheduler():
        sigma = self.sigmas[timestep_id]
        sample = (1 - sigma) * original_samples + sigma * noise
        return sample
+    
+
+    def training_target(self, sample, noise, timestep):
+        target = noise - sample
+        return target