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
+    
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