yhavinga commited on
Commit
2354f1b
β€’
1 Parent(s): 0ed2b71

Disable some models to reduce memory usage on HF spaces

Browse files
Files changed (1) hide show
  1. app.py +24 -24
app.py CHANGED
@@ -21,36 +21,36 @@ GENERATOR_LIST = [
21
  "task": TRANSLATION_EN_TO_NL,
22
  "split_sentences": False,
23
  },
24
- {
25
- "model_name": "yhavinga/ul2-large-en-nl",
26
- "desc": "UL2 large en->nl",
27
- "task": TRANSLATION_EN_TO_NL,
28
- "split_sentences": False,
29
- },
30
- {
31
- "model_name": "Helsinki-NLP/opus-mt-en-nl",
32
- "desc": "Opus MT en->nl",
33
- "task": TRANSLATION_EN_TO_NL,
34
- "split_sentences": True,
35
- },
36
- {
37
- "model_name": "Helsinki-NLP/opus-mt-nl-en",
38
- "desc": "Opus MT nl->en",
39
- "task": TRANSLATION_NL_TO_EN,
40
- "split_sentences": True,
41
- },
42
  {
43
  "model_name": "yhavinga/t5-small-24L-ccmatrix-multi",
44
  "desc": "T5 small nl24 ccmatrix nl-en",
45
  "task": TRANSLATION_NL_TO_EN,
46
  "split_sentences": True,
47
  },
48
- {
49
- "model_name": "yhavinga/byt5-small-ccmatrix-en-nl",
50
- "desc": "ByT5 small ccmatrix en->nl",
51
- "task": TRANSLATION_EN_TO_NL,
52
- "split_sentences": True,
53
- },
54
  # {
55
  # "model_name": "yhavinga/t5-base-36L-ccmatrix-multi",
56
  # "desc": "T5 base nl36 ccmatrix en->nl",
 
21
  "task": TRANSLATION_EN_TO_NL,
22
  "split_sentences": False,
23
  },
24
+ # {
25
+ # "model_name": "yhavinga/ul2-large-en-nl",
26
+ # "desc": "UL2 large en->nl",
27
+ # "task": TRANSLATION_EN_TO_NL,
28
+ # "split_sentences": False,
29
+ # },
30
+ # {
31
+ # "model_name": "Helsinki-NLP/opus-mt-en-nl",
32
+ # "desc": "Opus MT en->nl",
33
+ # "task": TRANSLATION_EN_TO_NL,
34
+ # "split_sentences": True,
35
+ # },
36
+ # {
37
+ # "model_name": "Helsinki-NLP/opus-mt-nl-en",
38
+ # "desc": "Opus MT nl->en",
39
+ # "task": TRANSLATION_NL_TO_EN,
40
+ # "split_sentences": True,
41
+ # },
42
  {
43
  "model_name": "yhavinga/t5-small-24L-ccmatrix-multi",
44
  "desc": "T5 small nl24 ccmatrix nl-en",
45
  "task": TRANSLATION_NL_TO_EN,
46
  "split_sentences": True,
47
  },
48
+ # {
49
+ # "model_name": "yhavinga/byt5-small-ccmatrix-en-nl",
50
+ # "desc": "ByT5 small ccmatrix en->nl",
51
+ # "task": TRANSLATION_EN_TO_NL,
52
+ # "split_sentences": True,
53
+ # },
54
  # {
55
  # "model_name": "yhavinga/t5-base-36L-ccmatrix-multi",
56
  # "desc": "T5 base nl36 ccmatrix en->nl",