Update app_lora1.py

app_lora1.py

@@ -100,17 +100,25 @@ def register_pipeline_feature(pipe, text: str):
 # =========================================================
 def log_pipe_calls(obj, obj_name="pipe"):
     for attr_name in dir(obj):
+        if attr_name.startswith("_"):
+            continue
         attr = getattr(obj, attr_name)
-        if callable(attr) and not attr_name.startswith("_"):
-            def make_wrapper(f, name):
-                def wrapper(*args, **kwargs):
-                    log(f"• {obj_name}.{name} called with args={args}, kwargs={kwargs}")
-                    return f(*args, **kwargs)
-                return wrapper
-            setattr(obj, attr_name, make_wrapper(attr, attr_name))
+
+        # Skip non-callables or torch modules
+        if not callable(attr) or isinstance(attr, torch.nn.Module):
+            continue
+
+        def make_wrapper(f, name):
+            def wrapper(*args, **kwargs):
+                log(f"• {obj_name}.{name} called with args={args}, kwargs={kwargs}")
+                return f(*args, **kwargs)
+            return wrapper
+
+        setattr(obj, attr_name, make_wrapper(attr, attr_name))
     return obj
 
 
+
 # =========================================================
 # DOWNLOAD SCRIPTS
 # =========================================================