LutaoJiang commited on
Commit
5be3795
·
1 Parent(s): f4015c9
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -172,16 +172,16 @@ def save_py3dmesh_with_trimesh_fast(meshes, save_glb_path=TEMP_MESH_ADDRESS, app
172
  @spaces.GPU
173
  def text_to_detailed(prompt, seed=None):
174
  # test nvdiffrast
175
- # subprocess.run(
176
- # shlex.split(
177
- # "rm -rf /home/user/.cache/torch_extensions/py310_cu121/nvdiffrast_plugin/lock"
178
- # )
179
- # )
180
  if "CUDA_VISIBLE_DEVICES" in os.environ:
181
  del os.environ["CUDA_VISIBLE_DEVICES"]
182
  subprocess.run(['nvidia-smi'])
183
  import nvdiffrast.torch as dr
184
- dr.RasterizeCudaContext(device="cuda:0")
185
  print("nvdiffrast initialized successfully")
186
 
187
  print(f"torch.cuda.is_available():{torch.cuda.is_available()}")
 
172
  @spaces.GPU
173
  def text_to_detailed(prompt, seed=None):
174
  # test nvdiffrast
175
+ subprocess.run(
176
+ shlex.split(
177
+ "rm -rf /home/user/.cache/torch_extensions/py310_cu121/nvdiffrast_plugin/lock"
178
+ )
179
+ )
180
  if "CUDA_VISIBLE_DEVICES" in os.environ:
181
  del os.environ["CUDA_VISIBLE_DEVICES"]
182
  subprocess.run(['nvidia-smi'])
183
  import nvdiffrast.torch as dr
184
+ dr.RasterizeCudaContext()
185
  print("nvdiffrast initialized successfully")
186
 
187
  print(f"torch.cuda.is_available():{torch.cuda.is_available()}")