Switch to local checkpoints

apps/mlm.py

@@ -43,7 +43,8 @@ def app(state):
     def load_model(ckpt):
         return FlaxCLIPVisionBertForMaskedLM.from_pretrained(ckpt)
 
-    mlm_checkpoints = ["flax-community/clip-vision-bert-cc12m-70k"]
+    #mlm_checkpoints = ["flax-community/clip-vision-bert-cc12m-70k"]
+    mlm_checkpoints = ["./ckpt/mlm/ckpt-60k"]
     dummy_data = pd.read_csv("cc12m_data/vqa_val.tsv", sep="\t")
 
     first_index = 15

apps/vqa.py

@@ -45,9 +45,10 @@ def app(state):
     def load_model(ckpt):
         return FlaxCLIPVisionBertForSequenceClassification.from_pretrained(ckpt)
 
-    vqa_checkpoints = [
-        "flax-community/clip-vision-bert-vqa-ft-6k"
-    ]  # TODO: Maybe add more checkpoints?
+    #vqa_checkpoints = [
+    #    "flax-community/clip-vision-bert-vqa-ft-6k"
+    #]  # TODO: Maybe add more checkpoints?
+    vqa_checkpoints = ["./ckpt/vqa/ckpt-60k-5999"]
     dummy_data = pd.read_csv("dummy_vqa_multilingual.tsv", sep="\t")
     code_to_name = {
         "en": "English",