Fgdfgfthgr
/

Anime_Images_Style_Embedder

Model card Files Files and versions

Fgdfgfthgr commited on Aug 2, 2025

Commit

da8234d

·

verified ·

1 Parent(s): 3d4d850

Update gallery_review.py

Files changed (1) hide show

gallery_review.py +1 -1

gallery_review.py CHANGED Viewed

@@ -317,7 +317,7 @@ if __name__ == '__main__':
     #folder = 'images_for_style_embedding'
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model = PLModule()
-    state_dict = load_file("Style_Embedder_v2.safetensors")
     model.network.load_state_dict(state_dict)
     # 'Grouping' or 'Explore'
     generate_embeddings(folder, 'Grouping', model)

     #folder = 'images_for_style_embedding'
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model = PLModule()
+    state_dict = load_file("Style_Embedder_v3.safetensors")
     model.network.load_state_dict(state_dict)
     # 'Grouping' or 'Explore'
     generate_embeddings(folder, 'Grouping', model)