diff --git a/CHANGELOG.md b/CHANGELOG.md index a5182571d..fef78d91c 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -46,7 +46,8 @@ - fix **RunPod** memory limit reporting - fix flux ipadapter with start/stop values - fix progress api `eta_relative` - - fix `insightface` loader + - fix `insightface` loader + - fix remove vae for flux.1 - **IPEX** - add `--upgrade` to torch_command when using `--use-nightly` for *ipex* and *rocm* - add xpu to profiler diff --git a/modules/sd_vae_remote.py b/modules/sd_vae_remote.py index 55b8aa7ff..c3591af7b 100644 --- a/modules/sd_vae_remote.py +++ b/modules/sd_vae_remote.py @@ -42,7 +42,9 @@ def remote_decode(latents: torch.Tensor, width: int = 0, height: int = 0, model_ for i in range(latents.shape[0]): try: - latent = latents[i].detach().clone().to(device=devices.cpu, dtype=devices.dtype).unsqueeze(0) + latent = latents[i].detach().clone().to(device=devices.cpu, dtype=devices.dtype) + if model_type != 'f1': + latent = latent.unsqueeze(0) params = { "input_tensor_type": "binary", "shape": list(latent.shape), @@ -76,7 +78,7 @@ def remote_decode(latents: torch.Tensor, width: int = 0, height: int = 0, model_ timeout=300, ) if not response.ok: - shared.log.error(f'Decode: type="remote" model={model_type} code={response.status_code} headers={response.headers} {response.json()}') + shared.log.error(f'Decode: type="remote" model={model_type} code={response.status_code} shape={latent.shape} url="{url}" args={params} headers={response.headers} response={response.json()}') else: content += len(response.content) if shared.opts.remote_vae_type == 'raw':