Tennineee commited on
Commit
4348c66
·
verified ·
1 Parent(s): fe1ae81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ DAMV2_configs = {
53
  'vitb': {'encoder': 'vitb', 'features': 128, 'out_channels': [96, 192, 384, 768]},
54
  'vitl': {'encoder': 'vitl', 'features': 256, 'out_channels': [256, 512, 1024, 1024]},
55
  }
56
- encoder = 'vits' # or 'vits', 'vitb', 'vitl'
57
  encoder2name = {
58
  'vits': 'Small',
59
  'vitb': 'Base',
 
53
  'vitb': {'encoder': 'vitb', 'features': 128, 'out_channels': [96, 192, 384, 768]},
54
  'vitl': {'encoder': 'vitl', 'features': 256, 'out_channels': [256, 512, 1024, 1024]},
55
  }
56
+ encoder = 'vitb' # or 'vits', 'vitb', 'vitl'
57
  encoder2name = {
58
  'vits': 'Small',
59
  'vitb': 'Base',