change output

inference/utils.py

@@ -168,7 +168,7 @@ def load_sd_upscale(ckpt, inf_device):
     return out
 
 
-def upscale(upscale_model, tensor: torch.Tensor, inf_device, output_device="cpu", upscale_factor) -> torch.Tensor:
+def upscale(upscale_model, tensor: torch.Tensor, inf_device, upscale_factor, output_device="cpu") -> torch.Tensor:
 
     memory_required = module_size(upscale_model.model)
     memory_required += (
@@ -201,11 +201,12 @@ def upscale(upscale_model, tensor: torch.Tensor, inf_device, output_device="cpu"
     return s
 
 
+
 def upscale_batch_and_concatenate(upscale_model, latents, inf_device, output_device="cpu", upscale_factor = 4) -> torch.Tensor:
     upscaled_latents = []
     for i in range(latents.size(0)):
         latent = latents[i]
-        upscaled_latent = upscale(upscale_model, latent, inf_device, output_device, upscale_factor)
+        upscaled_latent = upscale(upscale_model, latent, inf_device, upscale_factor, output_device)
         upscaled_latents.append(upscaled_latent)
     return torch.stack(upscaled_latents)