doc

Author: xadupre

CHANGELOGS.rst

@@ -4,6 +4,7 @@ Change Logs
 0.7.2
 +++++
 
+* :pr:`165`: support for task text-to-image
 * :pr:`162`: improves graphs rendering for historical data
 
 0.7.1

_doc/api/tasks/index.rst

@@ -46,6 +46,7 @@ Or:
     summarization
     text_classification
     text_generation
+    text_to_image
     text2text_generation
     zero_shot_image_classification
 

_doc/api/tasks/text_to_image.rst

@@ -0,0 +1,7 @@
+
+onnx_diagnostic.tasks.text_to_image
+===================================
+
+.. automodule:: onnx_diagnostic.tasks.text_to_image
+    :members:
+    :no-undoc-members:

onnx_diagnostic/tasks/text_to_image.py

@@ -41,7 +41,7 @@ def get_inputs(
     assert (
         "cls_cache" not in kwargs
     ), f"Not yet implemented for cls_cache={kwargs['cls_cache']!r}."
-    batch = torch.export.Dim("batch", min=1, max=1024)
+    batch = "batch"
     shapes = {
         "sample": {0: batch},
         "timestep": {},

onnx_diagnostic/torch_models/validate.py

@@ -541,7 +541,7 @@ def validate_model(
     summary["model_config"] = str(
         shrink_config(
             data["configuration"]
-            if type(data["configuration"])
+            if type(data["configuration"]) is dict
             else data["configuration"].to_dict()
         )
     ).replace(" ", "")