Update for separate CausalLM tests
Browse files
app.py
CHANGED
|
@@ -17,6 +17,7 @@ sys.path.append(str(test_path))
|
|
| 17 |
|
| 18 |
from test_modeling import *
|
| 19 |
from test_diffusion import *
|
|
|
|
| 20 |
|
| 21 |
RESULT_FILE = "supported_models.md"
|
| 22 |
|
|
@@ -30,7 +31,9 @@ def generate_model_list():
|
|
| 30 |
|
| 31 |
for test in tests:
|
| 32 |
task = test.replace("IntegrationTest", "").replace("Test", "")
|
| 33 |
-
if "
|
|
|
|
|
|
|
| 34 |
cls = getattr(import_module("test_modeling"), test)
|
| 35 |
else:
|
| 36 |
cls = getattr(import_module("test_diffusion"), test)
|
|
@@ -71,6 +74,5 @@ demo = gr.Interface(
|
|
| 71 |
title="List of validated architectures for optimum[openvino]",
|
| 72 |
inputs=[],
|
| 73 |
outputs=[gr.Markdown(), gr.File()],
|
| 74 |
-
allow_flagging="never",
|
| 75 |
)
|
| 76 |
demo.launch(server_name="0.0.0.0")
|
|
|
|
| 17 |
|
| 18 |
from test_modeling import *
|
| 19 |
from test_diffusion import *
|
| 20 |
+
from test_decoder import *
|
| 21 |
|
| 22 |
RESULT_FILE = "supported_models.md"
|
| 23 |
|
|
|
|
| 31 |
|
| 32 |
for test in tests:
|
| 33 |
task = test.replace("IntegrationTest", "").replace("Test", "")
|
| 34 |
+
if "OVModelForCausalLM" in task:
|
| 35 |
+
cls = getattr(import_module("test_decoder"), test)
|
| 36 |
+
elif "OVModelFor" in task:
|
| 37 |
cls = getattr(import_module("test_modeling"), test)
|
| 38 |
else:
|
| 39 |
cls = getattr(import_module("test_diffusion"), test)
|
|
|
|
| 74 |
title="List of validated architectures for optimum[openvino]",
|
| 75 |
inputs=[],
|
| 76 |
outputs=[gr.Markdown(), gr.File()],
|
|
|
|
| 77 |
)
|
| 78 |
demo.launch(server_name="0.0.0.0")
|