more diffusers work

This commit is contained in:
Vladimir Mandic
2023-05-27 15:49:54 -04:00
parent 24d8570bcb
commit 851d129680
6 changed files with 155 additions and 81 deletions
@@ -1,6 +1,5 @@
"""SAMPLING ONLY."""
import numpy as np
import torch
from .uni_pc import NoiseScheduleVP, model_wrapper, UniPC, get_time_steps