Spaces:

nyanko7
/

sd-diffusers-webui

Running on T4

nyanko7 commited on May 1, 2023

Commit

6614219

•

1 Parent(s): 6477781

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -242,7 +242,7 @@ def inference(
                 loaded_learned_embeds = torch.load(file, map_location="cpu")
             else:
                 loaded_learned_embeds = load_file(file, device="cpu")
-            loaded_learned_embeds = loaded_learned_embeds["string_to_param"]["*"] if "string_to_param" in loaded_learned_embed else loaded_learned_embed
             ti_embs[name] = loaded_learned_embeds
         if len(ti_embs) > 0:

                 loaded_learned_embeds = torch.load(file, map_location="cpu")
             else:
                 loaded_learned_embeds = load_file(file, device="cpu")
+            loaded_learned_embeds = loaded_learned_embeds["string_to_param"]["*"] if "string_to_param" in loaded_learned_embeds else loaded_learned_embeds
             ti_embs[name] = loaded_learned_embeds
         if len(ti_embs) > 0: