nodes/city96-SD-Latent-Upscaler
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Change confusing input name

Browse Source
This commit is contained in:
City
2023-08-15 20:24:21 +02:00
parent b4547b3386
commit aae69df386
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfy_latent_upscaler.py
+4 -4
View File
@@ -41,7 +41,7 @@ class LatentUpscaler:
return {
"required": {
"samples": ("LATENT", ),
"model_ver": (["v1", "xl"],),
"latent_ver": (["v1", "xl"],),
"scale_factor": (["1.25", "1.5", "2.0"],),
}
}
@@ -50,13 +50,13 @@ class LatentUpscaler:
FUNCTION = "upscale"
CATEGORY = "latent"
def upscale(self, samples, model_ver, scale_factor):
def upscale(self, samples, latent_ver, scale_factor):
model = Upscaler(scale_factor)
weights = str(hf_hub_download(
repo_id="city96/SD-Latent-Upscaler",
filename=f"latent-upscaler-v{model.version}_SD{model_ver}-x{scale_factor}.safetensors")
filename=f"latent-upscaler-v{model.version}_SD{latent_ver}-x{scale_factor}.safetensors")
)
# weights = f"./latent-upscaler-v{model.version}_SD{model_ver}-x{scale_factor}.safetensors"
# weights = f"./latent-upscaler-v{model.version}_SD{latent_ver}-x{scale_factor}.safetensors"
model.load_state_dict(load_file(weights))
lt = samples["samples"]