Update models/tag2text.py

models/tag2text.py

@@ -36,8 +36,7 @@ class Tag2Text_Caption(nn.Module):
                  vit_grad_ckpt = False,
                  vit_ckpt_layer = 0,
                  prompt = 'a picture of ',
-                 config = None,
-                 threshold = 0.2,
+                 threshold = 0.7,
                  ):
         """
         Args: