Hugging Face's logo

Spaces:
clarin-knext
/
dialogue-state-tracking
Runtime error

App Files Community
1
kwojtasik commited on
Commit
46848cd
·
1 Parent(s): 92fb351

Update models.py

Browse files

Leave one model for testing

Files changed (1) hide show
  1. models.py +30 -30
models.py CHANGED
@@ -22,41 +22,41 @@ DEFAULT_DST_INPUTS: Dict[str, str] = {
22
 
23
 
24
  DST_MODELS: Dict[str, Dict[str, Any]] = {
25
- "plt5-small": {
26
- "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-small-dst", use_auth_token=auth_token),
27
- "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-small-dst", use_auth_token=auth_token),
28
- "default_input": DEFAULT_DST_INPUTS["polish"],
29
- },
30
- "plt5-base": {
31
- "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-base-dst", use_auth_token=auth_token),
32
- "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-base-dst", use_auth_token=auth_token),
33
- "default_input": DEFAULT_DST_INPUTS["polish"],
34
- },
35
  "plt5-base-poquad-dst-v2": {
36
  "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-base-poquad-dst-v2", use_auth_token=auth_token),
37
  "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-base-poquad-dst-v2", use_auth_token=auth_token),
38
  "default_input": DEFAULT_DST_INPUTS["polish"],
39
  },
40
- "t5-small": {
41
- "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/t5-small-dst", use_auth_token=auth_token),
42
- "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/t5-small-dst", use_auth_token=auth_token),
43
- "default_input": DEFAULT_DST_INPUTS["english"],
44
- },
45
- "t5-base": {
46
- "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/t5-base-dst", use_auth_token=auth_token),
47
- "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/t5-base-dst", use_auth_token=auth_token),
48
- "default_input": DEFAULT_DST_INPUTS["english"],
49
- },
50
- "flant5-small [EN/PL]": {
51
- "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/flant5-small-dst", use_auth_token=auth_token),
52
- "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/flant5-small-dst", use_auth_token=auth_token),
53
- "default_input": DEFAULT_DST_INPUTS["english"],
54
- },
55
- "flant5-base [EN/PL]": {
56
- "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/flant5-base-dst", use_auth_token=auth_token),
57
- "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/flant5-base-dst", use_auth_token=auth_token),
58
- "default_input": DEFAULT_DST_INPUTS["english"],
59
- },
60
  }
61
 
62
 
 
22
 
23
 
24
  DST_MODELS: Dict[str, Dict[str, Any]] = {
25
+ # "plt5-small": {
26
+ # "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-small-dst", use_auth_token=auth_token),
27
+ # "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-small-dst", use_auth_token=auth_token),
28
+ # "default_input": DEFAULT_DST_INPUTS["polish"],
29
+ # },
30
+ # "plt5-base": {
31
+ # "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-base-dst", use_auth_token=auth_token),
32
+ # "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-base-dst", use_auth_token=auth_token),
33
+ # "default_input": DEFAULT_DST_INPUTS["polish"],
34
+ # },
35
  "plt5-base-poquad-dst-v2": {
36
  "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-base-poquad-dst-v2", use_auth_token=auth_token),
37
  "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-base-poquad-dst-v2", use_auth_token=auth_token),
38
  "default_input": DEFAULT_DST_INPUTS["polish"],
39
  },
40
+ # "t5-small": {
41
+ # "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/t5-small-dst", use_auth_token=auth_token),
42
+ # "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/t5-small-dst", use_auth_token=auth_token),
43
+ # "default_input": DEFAULT_DST_INPUTS["english"],
44
+ # },
45
+ # "t5-base": {
46
+ # "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/t5-base-dst", use_auth_token=auth_token),
47
+ # "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/t5-base-dst", use_auth_token=auth_token),
48
+ # "default_input": DEFAULT_DST_INPUTS["english"],
49
+ # },
50
+ # "flant5-small [EN/PL]": {
51
+ # "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/flant5-small-dst", use_auth_token=auth_token),
52
+ # "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/flant5-small-dst", use_auth_token=auth_token),
53
+ # "default_input": DEFAULT_DST_INPUTS["english"],
54
+ # },
55
+ # "flant5-base [EN/PL]": {
56
+ # "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/flant5-base-dst", use_auth_token=auth_token),
57
+ # "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/flant5-base-dst", use_auth_token=auth_token),
58
+ # "default_input": DEFAULT_DST_INPUTS["english"],
59
+ # },
60
  }
61
 
62