serenarolloh commited on
Commit
2c63dd8
·
verified ·
1 Parent(s): 545d7ce

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +6 -6
config.py CHANGED
@@ -34,14 +34,14 @@ class Settings(BaseSettings):
34
  # You had "donut-16" in some places and "donut-v16" in others
35
  shipper_model_map = {
36
  # Group 1 models - using donut-v16
37
- "61": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
38
- "81": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
39
- "139": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
40
 
41
  # Group 2 models - using donut-v17
42
- "165": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
43
- "127": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
44
- "145": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
45
  }
46
 
47
  previous_model = self.model
 
34
  # You had "donut-16" in some places and "donut-v16" in others
35
  shipper_model_map = {
36
  # Group 1 models - using donut-v16
37
+ 61: {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
38
+ 81: {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
39
+ 139: {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
40
 
41
  # Group 2 models - using donut-v17
42
+ 165: {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
43
+ 127: {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
44
+ 145: {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
45
  }
46
 
47
  previous_model = self.model