Fix inference command clip-roberta (#31)

huggingface · Feb 19, 2024 · 763f609 · 763f609
1 parent 92e4f64
commit 763f609
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/examples/contrastive-image-text/README.md b/examples/contrastive-image-text/README.md
@@ -250,5 +250,8 @@ python run_clip.py \
     --use_lazy_mode \
     --use_hpu_graphs_for_inference \
     --gaudi_config_name Habana/clip \
-    --bf16
+    --bf16 \
+    --mediapipe_dataloader
 ```
+
+> `--mediapipe_dataloader` only works on Gaudi2.