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 |
)
|