recognize-anything

Running

xinyu1205 commited on Mar 10, 2023

Commit

b0832a1

1 Parent(s): ae43413

Update models/tag2text.py

Files changed (1) hide show

models/tag2text.py CHANGED Viewed

@@ -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,
                  ):
         """
         Args:

                  vit_grad_ckpt = False,
                  vit_ckpt_layer = 0,
                  prompt = 'a picture of ',
+                 threshold = 0.7,
                  ):
         """
         Args: