Also use name property of loaded embedding .pt just like image embedd…

…ings
R-N · Nov 3, 2022 · 157907c · 157907c
1 parent efd4042
commit 157907c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py
@@ -99,6 +99,7 @@ def process_file(path, filename):
                     name = data.get('name', name)
             else:
                 data = torch.load(path, map_location="cpu")
+                name = data.get('name', name)
 
             # textual inversion embeddings
             if 'string_to_param' in data: