Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
move canonical checkpoints to their new location to keep this Space in "linked Spaces" (#1)
Browse files- move canonical checkpoints to their new location to keep this Space in "linked Spaces" (0db8483449542d22203aab2581ca490e514ad112)
- 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,
|