From 21f890f9da3cfbeaba8e2ac3c425ee9e998d5229 Mon Sep 17 00:00:00 2001
From: Robin Rombach <robin.rombach@gmail.com>
Date: Wed, 16 Nov 2022 21:34:06 +0100
Subject: [PATCH] Update sampler.py

fix import
---
 ldm/models/diffusion/dpm_solver/sampler.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ldm/models/diffusion/dpm_solver/sampler.py b/ldm/models/diffusion/dpm_solver/sampler.py
index 141414309..2c42d6f96 100644
--- a/ldm/models/diffusion/dpm_solver/sampler.py
+++ b/ldm/models/diffusion/dpm_solver/sampler.py
@@ -2,7 +2,7 @@
 
 import torch
 
-from .solver import NoiseScheduleVP, model_wrapper, DPM_Solver
+from .dpm_solver import NoiseScheduleVP, model_wrapper, DPM_Solver
 
 
 class DPMSolverSampler(object):
@@ -79,4 +79,4 @@ def sample(self,
         dpm_solver = DPM_Solver(model_fn, ns, predict_x0=True, thresholding=False)
         x = dpm_solver.sample(img, steps=S, skip_type="time_uniform", method="multistep", order=2, lower_order_final=True)
 
-        return x.to(device), None
\ No newline at end of file
+        return x.to(device), None