xinyu1205 commited on
Commit
6bb4894
1 Parent(s): 7185cc1

Update models/tag2text.py

Browse files
Files changed (1) hide show
  1. models/tag2text.py +1 -1
models/tag2text.py CHANGED
@@ -26,7 +26,7 @@ def read_json(rpath):
26
  with open(rpath, 'r') as f:
27
  return json.load(f)
28
 
29
- delete_tag_index = [135]
30
 
31
  class Tag2Text_Caption(nn.Module):
32
  def __init__(self,
 
26
  with open(rpath, 'r') as f:
27
  return json.load(f)
28
 
29
+ delete_tag_index = [127, 3351, 3265, 3338, 3355, 3359]
30
 
31
  class Tag2Text_Caption(nn.Module):
32
  def __init__(self,