justinj92 commited on
Commit
8624e9d
·
verified ·
1 Parent(s): 51a75e5

Update utils/models.py

Browse files
Files changed (1) hide show
  1. utils/models.py +1 -2
utils/models.py CHANGED
@@ -10,8 +10,7 @@ from transformers import AutoModelForCausalLM, AutoProcessor
10
  from utils.imports import fixed_get_imports
11
 
12
  CHECKPOINTS = [
13
- "microsoft/Florence-2-large-ft",
14
- "microsoft/Florence-2-base-ft",
15
  ]
16
 
17
 
 
10
  from utils.imports import fixed_get_imports
11
 
12
  CHECKPOINTS = [
13
+ "microsoft/Florence-2-large-ft"
 
14
  ]
15
 
16