debug

app.py

@@ -75,10 +75,6 @@ def check_gpu():
     subprocess.run(['nvidia-smi'])  # Test if CUDA is available
     print(f"torch.cuda.is_available:{torch.cuda.is_available()}")
     print("Device count:", torch.cuda.device_count())
-    # test nvdiffrast
-    import nvdiffrast.torch as dr
-    dr.RasterizeCudaContext(device="cuda:0")
-    print("nvdiffrast initialized successfully")
 
 # Only check GPU in non-UI debug mode
 if not UI_ONLY_MODE:
@@ -187,6 +183,11 @@ if UI_ONLY_MODE:
 else:
     @spaces.GPU
     def text_to_detailed(prompt, seed=None):
+        # test nvdiffrast
+        import nvdiffrast.torch as dr
+        dr.RasterizeCudaContext(device="cuda:0")
+        print("nvdiffrast initialized successfully")
+        
         print(f"torch.cuda.is_available():{torch.cuda.is_available()}")
         # print(f"Before text_to_detailed: {torch.cuda.memory_allocated() / 1024**3} GB")
         return k3d_wrapper.get_detailed_prompt(prompt, seed)