adamelliotfields commited on
Commit
4c34ed8
·
verified ·
1 Parent(s): 039ff6d

Fix upscaler unloading

Browse files
Files changed (1) hide show
  1. 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)