sunnychenxiwang commited on
Commit
bf18e31
·
verified ·
1 Parent(s): 11195f8

Update pipeline/GroundingDINO/groundingdino/util/get_tokenlizer.py

Browse files
pipeline/GroundingDINO/groundingdino/util/get_tokenlizer.py CHANGED
@@ -16,7 +16,7 @@ def get_tokenlizer(text_encoder_type):
16
  )
17
  print("final text_encoder_type: {}".format(text_encoder_type))
18
  # text_encoder_type
19
- tokenizer = AutoTokenizer.from_pretrained("/newdisk3/wcx/models--bert-base-uncased/snapshots/1dbc166cf8765166998eff31ade2eb64c8a40076")
20
  return tokenizer
21
 
22
 
 
16
  )
17
  print("final text_encoder_type: {}".format(text_encoder_type))
18
  # text_encoder_type
19
+ tokenizer = AutoTokenizer.from_pretrained(text_encoder_type)
20
  return tokenizer
21
 
22