From 7c64d8ed54a543851ddf1c07d6fa306a1a9be25b Mon Sep 17 00:00:00 2001 From: dankernel Date: Tue, 12 Sep 2023 13:42:03 +0900 Subject: [PATCH 1/2] Added: requirements.txt --- requirements.txt | 1 + 1 file changed, 1 insertion(+) create mode 100644 requirements.txt diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..4925907 --- /dev/null +++ b/requirements.txt @@ -0,0 +1 @@ +open_clip_torch From a1a759c6ae247c23c1ed3af494ee3a6e95689e6b Mon Sep 17 00:00:00 2001 From: dankernel Date: Tue, 12 Sep 2023 13:44:06 +0900 Subject: [PATCH 2/2] Fix: Added missing `pretrained` parameter --- compute_openclip_text_embeddings.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/compute_openclip_text_embeddings.py b/compute_openclip_text_embeddings.py index 022beb8..ada684a 100644 --- a/compute_openclip_text_embeddings.py +++ b/compute_openclip_text_embeddings.py @@ -30,6 +30,7 @@ parser.add_argument("text_prompts_file", type=str) parser.add_argument("output_path", type=str) parser.add_argument("--model_name", type=str, default="ViT-B-32") + parser.add_argument("--pretrained", type=str, default="laion2b_s34b_b79k") args = parser.parse_args() with open(args.text_prompts_file, 'r') as f: @@ -52,4 +53,4 @@ print(f"Saving text embeddings to {args.output_path}") np.save(args.output_path, text_embeddings) - \ No newline at end of file +