fix scheduler caching
This commit is contained in:
parent
18f59f034d
commit
cb8e9e7080
|
@ -215,6 +215,6 @@ def load_pipeline(
|
||||||
pipe = pipe.to(device.torch_str())
|
pipe = pipe.to(device.torch_str())
|
||||||
|
|
||||||
server.cache.set("diffusion", pipe_key, pipe)
|
server.cache.set("diffusion", pipe_key, pipe)
|
||||||
server.cache.set("scheduler", scheduler_key, scheduler)
|
server.cache.set("scheduler", scheduler_key, components["scheduler"])
|
||||||
|
|
||||||
return pipe
|
return pipe
|
||||||
|
|
Loading…
Reference in New Issue