yonikremer
commited on
Commit
·
818ab4f
1
Parent(s):
59660ba
added gpt2 to the models list for testing
Browse files- available_models.py +1 -0
available_models.py
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
AVAILABLE_MODELS = (
|
2 |
"facebook/opt-iml-max-1.3b",
|
3 |
"facebook/opt-iml-max-30b",
|
|
|
4 |
)
|
|
|
1 |
AVAILABLE_MODELS = (
|
2 |
"facebook/opt-iml-max-1.3b",
|
3 |
"facebook/opt-iml-max-30b",
|
4 |
+
"gpt2",
|
5 |
)
|