artem/stable-diffusion
1
0
Fork 0
mirror of https://github.com/CompVis/stable-diffusion.git synced 2026-04-02 07:02:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update sampler.py

Browse Source 
fix import
This commit is contained in:
Robin Rombach 2022-11-16 21:34:06 +01:00 committed by GitHub
parent 5a00c4f8db
commit 21f890f9da
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ldm/models/diffusion/dpm_solver/sampler.py
View File

@ -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):
@ -79,4 +79,4 @@ class DPMSolverSampler(object):
dpm_solver = DPM_Solver(model_fn, ns, predict_x0=True, thresholding=False) 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) x = dpm_solver.sample(img, steps=S, skip_type="time_uniform", method="multistep", order=2, lower_order_final=True)
return x.to(device), None return x.to(device), None