julien-c HF staff commited on
Commit
0db8483
1 Parent(s): 9ca4d8e

move canonical checkpoints to their new location to keep this Space in "linked Spaces"

Browse files

cc

@XciD



@lysandre

Files changed (1) hide show
  1. 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,