From 5f65e3c2bc83fb40d3a1d8dbead6922ef4b7b6b9 Mon Sep 17 00:00:00 2001
From: Mark Saroufim <marksaroufim@meta.com>
Date: Wed, 26 Jul 2023 00:18:04 +0000
Subject: [PATCH] xu feedbacK

---
 torchbenchmark/models/clip/metadata.yaml    | 4 ++--
 torchbenchmark/models/clip/requirements.txt | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/torchbenchmark/models/clip/metadata.yaml b/torchbenchmark/models/clip/metadata.yaml
index a17a4f9adb..6dab8abdab 100644
--- a/torchbenchmark/models/clip/metadata.yaml
+++ b/torchbenchmark/models/clip/metadata.yaml
@@ -1,7 +1,7 @@
 devices:
   NVIDIA A100-SXM4-40GB:
-    eval_batch_size: 1
-    eval_train_size: 1
+    eval_batch_size: 32
+    train_batch_size: 32
 eval_benchmark: false
 eval_deterministic: false
 eval_nograd: true
diff --git a/torchbenchmark/models/clip/requirements.txt b/torchbenchmark/models/clip/requirements.txt
index 63e9897066..86908bf5a6 100644
--- a/torchbenchmark/models/clip/requirements.txt
+++ b/torchbenchmark/models/clip/requirements.txt
@@ -1 +1 @@
-git+https://github.com/facebookresearch/multimodal.git
\ No newline at end of file
+git+https://github.com/facebookresearch/multimodal.git@bf42547c5fcfd063eba4f036def0c57309c2ee57
\ No newline at end of file