Spaces:
Running
on
Zero
Running
on
Zero
adamelliotfields
commited on
Fix upscaler unloading
Browse files- lib/loader.py +2 -2
lib/loader.py
CHANGED
@@ -300,7 +300,7 @@ class Loader:
|
|
300 |
# defaults to float32
|
301 |
pipe_kwargs["torch_dtype"] = torch.float16
|
302 |
|
303 |
-
self._unload(kind, model, ip_adapter, deepcache)
|
304 |
self._load_pipeline(kind, model, progress, **pipe_kwargs)
|
305 |
|
306 |
# error loading model
|
@@ -323,7 +323,7 @@ class Loader:
|
|
323 |
self.pipe.scheduler = Config.SCHEDULERS[scheduler](**scheduler_kwargs)
|
324 |
|
325 |
self._load_vae(taesd, model)
|
326 |
-
self._load_upscaler(scale)
|
327 |
self._load_freeu(freeu)
|
328 |
self._load_deepcache(deepcache)
|
329 |
self._load_ip_adapter(ip_adapter)
|
|
|
|
300 |
# defaults to float32
|
301 |
pipe_kwargs["torch_dtype"] = torch.float16
|
302 |
|
303 |
+
self._unload(kind, model, ip_adapter, deepcache, scale)
|
304 |
self._load_pipeline(kind, model, progress, **pipe_kwargs)
|
305 |
|
306 |
# error loading model
|
|
|
323 |
self.pipe.scheduler = Config.SCHEDULERS[scheduler](**scheduler_kwargs)
|
324 |
|
325 |
self._load_vae(taesd, model)
|
|
|
326 |
self._load_freeu(freeu)
|
327 |
self._load_deepcache(deepcache)
|
328 |
self._load_ip_adapter(ip_adapter)
|
329 |
+
self._load_upscaler(scale)
|