Fixed breaking change with diffusers. Allow flowmatch on normal stable diffusion models.

This commit is contained in:
Jaret Burkett
2024-08-22 14:36:22 -06:00
parent e07a98a50c
commit 338c77d677
4 changed files with 37 additions and 28 deletions

View File

@@ -8,6 +8,7 @@ import torch
class CustomFlowMatchEulerDiscreteScheduler(FlowMatchEulerDiscreteScheduler):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.init_noise_sigma = 1.0
with torch.no_grad():
# create weights for timesteps