Spaces:
Running
Running
Commit
·
f61b4e0
1
Parent(s):
4baba46
removed aragpt2 medium and large for ram reduction
Browse files- backend/aragpt.py +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)
|