Fix face hires with lowvram

This commit is contained in:
Disty0
2024-06-21 18:09:56 +03:00
parent bc0e45d730
commit 43a54317ee
2 changed files with 2 additions and 1 deletions
+1
View File
@@ -76,6 +76,7 @@ But there's more than SD3:
- fix api face-hires
- fix api ip-adapter
- fix memory exceptions with ROCm, thanks @Disty0!
- fix face-hires with lowvram, thanks @Disty0!
- cleanup image metadata
- restructure api examples: `cli/api-*`
- handle theme fallback when invalid theme is specified
+1 -1
View File
@@ -764,7 +764,7 @@ def move_model(model, device=None, force=False):
if model is None or device is None:
return
if getattr(model, 'vae', None) is not None and get_diffusers_task(model) != DiffusersTaskType.TEXT_2_IMAGE:
if device == devices.device: # force vae back to gpu if not in txt2img mode
if device == devices.device and not model.vae.device.type == "meta": # force vae back to gpu if not in txt2img mode
model.vae.to(device)
if hasattr(model.vae, '_hf_hook'):
debug_move(f'Model move: to={device} class={model.vae.__class__} fn={sys._getframe(1).f_code.co_name}') # pylint: disable=protected-access