Skip to content

Commit 7352f06

Browse files
committed
fix(agent): init .py missing DPM2/DPM++ exports.
1 parent 235e0f8 commit 7352f06

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

iddm/model/samples/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,3 +24,5 @@
2424
from .ddim import DDIMDiffusion
2525
from .ddpm import DDPMDiffusion
2626
from .plms import PLMSDiffusion
27+
from .dpm2 import DPM2Diffusion
28+
from .dpmpp import DPMPlusPlusDiffusion

0 commit comments

Comments
 (0)