added Marimo support for echo=True

Author: peterwilli

guidance/models/_model.py

@@ -29,7 +29,11 @@
     torch_is_imported = True
 except ImportError:
     torch_is_imported = False
-
+try:
+    import marimo
+    marimo_is_imported = True
+except ImportError:
+    marimo_is_imported = False
 
 logger = logging.getLogger(__name__)
 try:
@@ -1026,6 +1030,8 @@ def _update_display(self, throttle=True):
             if ipython_is_imported:
                 clear_output(wait=True)
                 display(HTML(self._html()))
+            elif marimo_is_imported:
+                marimo.output.replace(marimo.Html(self._html()))
             else:
                 pprint(self._state)
 
@@ -1612,4 +1618,4 @@ def _check_dominated(node, parser, match_version, next_byte_mask):
             parser.pos = curr_pos
             if not child_dominate:
                 return False
-    return True
+    return True