Update app_lora1.py

app_lora1.py

@@ -91,6 +91,7 @@ def register_scripts(selected_scripts):
 # =========================================================
 # GPU-ONLY PIPELINE BUILDER (CRITICAL)
 # =========================================================
+
 @spaces.GPU
 def get_pipeline(script_name):
     if script_name in PIPELINES:
@@ -99,12 +100,38 @@ def get_pipeline(script_name):
     log(f"🔧 Building pipeline from {script_name}")
 
     namespace = {
+        "__file__": script_name,
+        "__name__": "__main__",
+
+        # Core
         "torch": torch,
+
+        # Diffusers
         "DiffusionPipeline": DiffusionPipeline,
+
+        # Z-Image specific (imported lazily)
+        "ZImagePipeline": __import__(
+            "diffusers.pipelines.z_image.pipeline_z_image",
+            fromlist=["ZImagePipeline"],
+        ).ZImagePipeline,
+
+        "ZImageTransformer2DModel": __import__(
+            "diffusers.models.transformers.z_image_transformer_2d",
+            fromlist=["ZImageTransformer2DModel"],
+        ).ZImageTransformer2DModel,
+
+        # Quantization helpers
+        "GGUFQuantizationConfig": __import__(
+            "diffusers.quantizers.gguf",
+            fromlist=["GGUFQuantizationConfig"],
+        ).GGUFQuantizationConfig,
     }
 
-    # Execute script logic (this WILL touch CUDA)
-    exec(SCRIPT_CODE[script_name], namespace)
+    try:
+        exec(SCRIPT_CODE[script_name], namespace)
+    except Exception as e:
+        log(f"❌ Script failed: {script_name}")
+        raise RuntimeError(f"Pipeline build failed for {script_name}") from e
 
     if "pipe" not in namespace:
         raise RuntimeError(f"{script_name} did not define `pipe`")
@@ -115,6 +142,7 @@ def get_pipeline(script_name):
     return PIPELINES[script_name]
 
 
+
 # =========================================================
 # IMAGE GENERATION (LOGIC UNCHANGED)
 # =========================================================