helenai commited on
Commit
9ddbb00
·
1 Parent(s): cca5b01

Update for separate CausalLM tests

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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 "OVModelFor" in task:
 
 
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")