Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
move canonical checkpoints to their new location to keep this Space in "linked Spaces"
Browse files- backend/GPUHandler.py +7 -7
backend/GPUHandler.py
CHANGED
@@ -17,7 +17,7 @@ model_metadata = {
|
|
17 |
"tokenizer": OpenAIGPTTokenizer,
|
18 |
"model": OpenAIGPTLMHeadModel,
|
19 |
"size": 550,
|
20 |
-
"checkpoint": "openai-gpt",
|
21 |
"identifier": "gpt"
|
22 |
}, "xlnet": {
|
23 |
"tokenizer": XLNetTokenizer,
|
@@ -35,13 +35,13 @@ model_metadata = {
|
|
35 |
"tokenizer": GPT2Tokenizer,
|
36 |
"model": GPT2LMHeadModel,
|
37 |
"size": 1500,
|
38 |
-
"checkpoint": "gpt2-medium",
|
39 |
"identifier": "gpt2/medium"
|
40 |
}, "gpt2/large": {
|
41 |
"tokenizer": GPT2Tokenizer,
|
42 |
"model": GPT2LMHeadModel,
|
43 |
"size": 3300,
|
44 |
-
"checkpoint": "gpt2-large",
|
45 |
"identifier": "gpt2/large"
|
46 |
}, "distilgpt2/small": {
|
47 |
"tokenizer": GPT2Tokenizer,
|
@@ -53,25 +53,25 @@ model_metadata = {
|
|
53 |
"tokenizer": CTRLTokenizer,
|
54 |
"model": CTRLLMHeadModel,
|
55 |
"size": 6300,
|
56 |
-
"checkpoint": "ctrl",
|
57 |
"identifier": "ctrl"
|
58 |
}, "pplm": {
|
59 |
"tokenizer": GPT2Tokenizer,
|
60 |
"model": GPT2LMHeadModel,
|
61 |
"size": 3000,
|
62 |
-
"checkpoint": "gpt2-large",
|
63 |
"identifier": "pplm"
|
64 |
}, "gpt2/xl": {
|
65 |
"tokenizer": GPT2Tokenizer,
|
66 |
"model": GPT2LMHeadModel,
|
67 |
"size": 7000,
|
68 |
-
"checkpoint": "gpt2-xl",
|
69 |
"identifier": "gpt2/xl"
|
70 |
}, "pplm": {
|
71 |
"tokenizer": GPT2Tokenizer,
|
72 |
"model": GPT2LMHeadModel,
|
73 |
"size": 4000,
|
74 |
-
"checkpoint": "gpt2-medium",
|
75 |
"identifier": "pplm",
|
76 |
"configuration_options": {
|
77 |
"config": GPT2Config,
|
|
|
17 |
"tokenizer": OpenAIGPTTokenizer,
|
18 |
"model": OpenAIGPTLMHeadModel,
|
19 |
"size": 550,
|
20 |
+
"checkpoint": "openai-community/openai-gpt",
|
21 |
"identifier": "gpt"
|
22 |
}, "xlnet": {
|
23 |
"tokenizer": XLNetTokenizer,
|
|
|
35 |
"tokenizer": GPT2Tokenizer,
|
36 |
"model": GPT2LMHeadModel,
|
37 |
"size": 1500,
|
38 |
+
"checkpoint": "openai-community/gpt2-medium",
|
39 |
"identifier": "gpt2/medium"
|
40 |
}, "gpt2/large": {
|
41 |
"tokenizer": GPT2Tokenizer,
|
42 |
"model": GPT2LMHeadModel,
|
43 |
"size": 3300,
|
44 |
+
"checkpoint": "openai-community/gpt2-large",
|
45 |
"identifier": "gpt2/large"
|
46 |
}, "distilgpt2/small": {
|
47 |
"tokenizer": GPT2Tokenizer,
|
|
|
53 |
"tokenizer": CTRLTokenizer,
|
54 |
"model": CTRLLMHeadModel,
|
55 |
"size": 6300,
|
56 |
+
"checkpoint": "Salesforce/ctrl",
|
57 |
"identifier": "ctrl"
|
58 |
}, "pplm": {
|
59 |
"tokenizer": GPT2Tokenizer,
|
60 |
"model": GPT2LMHeadModel,
|
61 |
"size": 3000,
|
62 |
+
"checkpoint": "openai-community/gpt2-large",
|
63 |
"identifier": "pplm"
|
64 |
}, "gpt2/xl": {
|
65 |
"tokenizer": GPT2Tokenizer,
|
66 |
"model": GPT2LMHeadModel,
|
67 |
"size": 7000,
|
68 |
+
"checkpoint": "openai-community/gpt2-xl",
|
69 |
"identifier": "gpt2/xl"
|
70 |
}, "pplm": {
|
71 |
"tokenizer": GPT2Tokenizer,
|
72 |
"model": GPT2LMHeadModel,
|
73 |
"size": 4000,
|
74 |
+
"checkpoint": "openai-community/gpt2-medium",
|
75 |
"identifier": "pplm",
|
76 |
"configuration_options": {
|
77 |
"config": GPT2Config,
|