Update app.py

app.py

@@ -5,25 +5,33 @@ import threading
 from pathlib import Path
 
 # === Runtime environment tweaks (must be set before importing heavy libs) ===
-os.environ.setdefault("CUDA_VISIBLE_DEVICES", "")        # force CPU
-os.environ.setdefault("WORLDGEN_DISABLE_EVAL", "1")       # avoid KNN/eval paths if supported
+os.environ.setdefault("CUDA_VISIBLE_DEVICES", "")  # force CPU
+os.environ.setdefault("WORLDGEN_DISABLE_EVAL", "1")  # avoid KNN/eval paths if supported
 os.environ.setdefault("GRADIO_TEMP_DIR", "/tmp/gradio")  # centralize gradio cache
 
 import nunchaku.utils
-def safe_get_precision():
-    import torch
-    if not torch.cuda.is_available():
-        return 'fp32'
-    return nunchaku.utils.get_precision()
+
+def safe_get_precision(*args, **kwargs):
+    # Always return fp32 on CPU to avoid any torch.cuda references
+    return "fp32"
+
+# Patch module attribute
 nunchaku.utils.get_precision = safe_get_precision
 
+# ALSO patch the module that imported the symbol name directly
+try:
+    import nunchaku.models.transformers.transformer_flux as tf
+    tf.get_precision = safe_get_precision
+except Exception:
+    # If the module isn't imported yet, it will still use our utils patch when it imports later.
+    pass
+
 import gradio as gr
 import torch
 from worldgen import WorldGen
 
 # Lazy init to avoid long cold start
 _wg = None
-
 def get_worldgen():
     global _wg
     if _wg is None: