mirror of
https://github.com/CompVis/stable-diffusion.git
synced 2026-04-09 10:23:35 +00:00
Update sampler.py
fix import
This commit is contained in:
parent
5a00c4f8db
commit
21f890f9da
@ -2,7 +2,7 @@
|
|||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
|
||||||
from .solver import NoiseScheduleVP, model_wrapper, DPM_Solver
|
from .dpm_solver import NoiseScheduleVP, model_wrapper, DPM_Solver
|
||||||
|
|
||||||
|
|
||||||
class DPMSolverSampler(object):
|
class DPMSolverSampler(object):
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user