support *.safetensors,*.bin embeddings

* EasyNegative.safetensors * ulzzang-6500-v1.1.bin
UnstableHuman · May 26, 2023 · 2e97ecf · 2e97ecf
1 parent 68a419d
commit 2e97ecf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/civitai/lib.py b/civitai/lib.py
@@ -206,7 +206,7 @@ def load_resource_list(types=['LORA', 'LoCon', 'Hypernetwork', 'TextualInversion
         resources += get_resources_in_folder('Hypernetwork', shared.cmd_opts.hypernetwork_dir, ['pt', 'safetensors', 'ckpt'])
     if 'TextualInversion' in types:
         resources = [r for r in resources if r['type'] != 'TextualInversion']
-        resources += get_resources_in_folder('TextualInversion', shared.cmd_opts.embeddings_dir, ['pt'])
+        resources += get_resources_in_folder('TextualInversion', shared.cmd_opts.embeddings_dir, ['pt', 'bin', 'safetensors'])
     if 'Checkpoint' in types:
         resources = [r for r in resources if r['type'] != 'Checkpoint']
         resources += get_resources_in_folder('Checkpoint', sd_models.model_path, ['safetensors', 'ckpt'], ['vae.safetensors', 'vae.ckpt'])