wissamantoun commited on
Commit
f61b4e0
1 Parent(s): 4baba46

removed aragpt2 medium and large for ram reduction

Browse files
Files changed (2) hide show
  1. backend/aragpt.py +2 -2
  2. backend/services.py +2 -2
backend/aragpt.py CHANGED
@@ -37,8 +37,8 @@ def write():
37
  "Model Selector",
38
  options=[
39
  "AraGPT2-Base",
40
- "AraGPT2-Medium",
41
- "Aragpt2-Large",
42
  "AraGPT2-Mega",
43
  ],
44
  index=0,
 
37
  "Model Selector",
38
  options=[
39
  "AraGPT2-Base",
40
+ # "AraGPT2-Medium",
41
+ # "Aragpt2-Large",
42
  "AraGPT2-Mega",
43
  ],
44
  index=0,
backend/services.py CHANGED
@@ -46,8 +46,8 @@ class TextGeneration:
46
  # }
47
  self.model_names_or_paths = {
48
  "aragpt2-medium": "aubmindlab/aragpt2-medium",
49
- "aragpt2-base": "aubmindlab/aragpt2-base",
50
- "aragpt2-large": "aubmindlab/aragpt2-large",
51
  "aragpt2-mega": "aubmindlab/aragpt2-mega",
52
  }
53
  set_seed(42)
 
46
  # }
47
  self.model_names_or_paths = {
48
  "aragpt2-medium": "aubmindlab/aragpt2-medium",
49
+ # "aragpt2-base": "aubmindlab/aragpt2-base",
50
+ # "aragpt2-large": "aubmindlab/aragpt2-large",
51
  "aragpt2-mega": "aubmindlab/aragpt2-mega",
52
  }
53
  set_seed(42)