new

app.py

@@ -154,7 +154,8 @@ def text_to_image(prompt, seed=None, strength=1.0,lora_scale=1.0, num_inference_
     # if init_image_path is not None:
     #     init_image = Image.open(init_image_path)
     subprocess.run(['nvidia-smi'])  # 测试 CUDA 是否可用
-    result = k3d_wrapper.generate_3d_bundle_image_text( 
+    with torch.no_grad():
+        result = k3d_wrapper.generate_3d_bundle_image_text( 
                                       prompt,
                                       image=init_image, 
                                       strength=strength,