diff --git a/backend/modules/k_diffusion_extra.py b/backend/modules/k_diffusion_extra.py index 4d9f4693..945871c2 100644 --- a/backend/modules/k_diffusion_extra.py +++ b/backend/modules/k_diffusion_extra.py @@ -1,6 +1,7 @@ # Only include samplers that are not already in A1111 import torch + from tqdm import trange