Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix remote vae for flux

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-03-16 12:42:13 -04:00
parent 86e55ef11d
commit 2c726c1d5e
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_vae_remote.py
+4 -2
View File
@@ -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':