AbstractPhil commited on
Commit
78742f1
·
verified ·
1 Parent(s): 1d3821a

Update 1model_cascade.py

Browse files
Files changed (1) hide show
  1. 1model_cascade.py +2 -0
1model_cascade.py CHANGED
@@ -2,6 +2,8 @@ from diffusers import DDPMScheduler
2
  import torch, math
3
 
4
  # doesn't work like the multimodel scheduler, but it's a fair substitute - alebeit missing some noise functions for this one
 
 
5
  class SurgeScheduler(DDPMScheduler):
6
  def __init__(self, *args, snr_gamma: float = 5.0, noise_offset: float = 0.1, **kwargs):
7
  super().__init__(
 
2
  import torch, math
3
 
4
  # doesn't work like the multimodel scheduler, but it's a fair substitute - alebeit missing some noise functions for this one
5
+ # it's highly basic and doesn't fit the correct paradigm to train sd15 correctly from 0, which is why I didn't bother posting it until now
6
+ # there are more robust iterations that will need fitting to create the necessary multi-model scheduler
7
  class SurgeScheduler(DDPMScheduler):
8
  def __init__(self, *args, snr_gamma: float = 5.0, noise_offset: float = 0.1, **kwargs):
9
  super().__init__(