mirror of
https://github.com/CompVis/stable-diffusion.git
synced 2026-04-02 07:02:32 +00:00
Update sampler.py
fix import
This commit is contained in:
parent
5a00c4f8db
commit
21f890f9da
@ -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):
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user