Skip to content

Commit 158e709

Browse files
committed
remove unusued endpoints
1 parent 34a0ecf commit 158e709

File tree

3 files changed

+6
-7
lines changed

3 files changed

+6
-7
lines changed

open/core/tasks.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@ def check_services_running():
2323

2424
endpoints = [
2525
settings.GPT2_MEDIUM_API_ENDPOINT,
26-
settings.GPT2_SMALL_API_ENDPOINT,
2726
settings.GPT2_LARGE_API_ENDPOINT,
2827
settings.GPT2_MEDIUM_GOT_API_ENDPOINT,
2928
settings.GPT2_MEDIUM_HP_API_ENDPOINT,

open/core/writeup/constants.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -54,10 +54,10 @@ class MLModelNames:
5454
XLNET_LARGE_CASED = "xlnet-large-cased"
5555
GPT2_MEDIUM = "gpt2-medium"
5656
TRANSFO_XL_WT103 = "transfo-xl-wt103"
57-
GPT2_SMALL = "gpt2"
57+
# GPT2_SMALL = "gpt2"
5858
GPT2_LARGE = "gpt2-large"
5959
# all the new self-generated
60-
GPT2_SMALL_LEGAL = "gpt2-small-legal"
60+
# GPT2_SMALL_LEGAL = "gpt2-small-legal"
6161
GPT2_MEDIUM_LEGAL = "gpt2-medium-legal"
6262
GPT2_MEDIUM_HP = "gpt2-medium-hp"
6363
GPT2_MEDIUM_GOT = "gpt2-medium-got"
@@ -69,8 +69,8 @@ class MLModelNames:
6969
ML_MODEL_NAME_CHOICES = [
7070
(MLModelNames.XLNET_LARGE_CASED, "XLNet Large Cased"),
7171
(MLModelNames.XLNET_BASE_CASED, "XLNet Base Cased"),
72-
(MLModelNames.GPT2_SMALL, "GPT2 Small"),
73-
(MLModelNames.GPT2_SMALL_LEGAL, "GPT2 Small Legal"),
72+
# (MLModelNames.GPT2_SMALL, "GPT2 Small"),
73+
# (MLModelNames.GPT2_SMALL_LEGAL, "GPT2 Small Legal"),
7474
(MLModelNames.GPT2_MEDIUM, "GPT2 Medium"),
7575
(MLModelNames.GPT2_MEDIUM_LEGAL, "GPT2 Medium Legal"),
7676
(MLModelNames.GPT2_MEDIUM_HP, "GPT2 Medium HP"),

open/core/writeup/consumers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,8 +76,8 @@ def get_api_endpoint_from_model_name(model_name):
7676
MLModelNames.GPT2_MEDIUM_LYRICS: settings.GPT2_MEDIUM_LYRICS_API_ENDPOINT,
7777
MLModelNames.GPT2_MEDIUM_COMPANIES: settings.GPT2_MEDIUM_COMPANIES_API_ENDPOINT,
7878
MLModelNames.GPT2_LARGE: settings.GPT2_LARGE_API_ENDPOINT,
79-
MLModelNames.GPT2_SMALL: settings.GPT2_SMALL_API_ENDPOINT,
80-
MLModelNames.GPT2_SMALL_LEGAL: settings.GPT2_SMALL_LEGAL_API_ENDPOINT,
79+
# MLModelNames.GPT2_SMALL: settings.GPT2_SMALL_API_ENDPOINT,
80+
# MLModelNames.GPT2_SMALL_LEGAL: settings.GPT2_SMALL_LEGAL_API_ENDPOINT,
8181
MLModelNames.XLNET_BASE_CASED: settings.XLNET_BASE_CASED_API_ENDPOINT,
8282
MLModelNames.XLNET_LARGE_CASED: settings.XLNET_LARGE_CASED_API_ENDPOINT,
8383
MLModelNames.TRANSFO_XL_WT103: settings.TRANSFORMERS_XL_API_ENDPOINT,

0 commit comments

Comments
 (0)