Fix missing import for random

2024-02-02 09:52:22 -06:00
parent 6e56dfe04d
commit 6c0f6ddf22
1 changed files with 1 additions and 0 deletions
@@ -10,6 +10,7 @@ import comfy.sample

 import k_diffusion.sampling
 from k_diffusion.sampling import BrownianTreeNoiseSampler, PIDStepSizeController, get_ancestral_step, to_d, default_noise_sampler
+import random

 # The following function adds the samplers during initialization, in __init__.py
 def add_samplers():