poor results

2026-06-27 18:06:19 +08:00 · 2022-12-24 07:37:19 +08:00
parent 6c84953ac1
commit 204aed2371
2 changed files with 572 additions and 20 deletions
@@ -482,7 +482,7 @@ class TimeGradPredictionNetwork2(TimeGradTrainingNetwork2):

        repeated_scale = repeat(scale)
        if self.scaling:
-            self.diffusion.scale = repeated_scale
+            self.diffusion.scale = repeated_scale.permute(0, 2, 1)[:1]
            
        future_samples = []
        for _ in range(self.num_parallel_samples):