LTT commited on
Commit
37052d4
·
verified ·
1 Parent(s): 35f45a3
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -193,7 +193,8 @@ def text_to_detailed(prompt, seed=None):
193
  )
194
  )
195
  import nvdiffrast.torch as dr
196
- dr.RasterizeCudaContext()
 
197
  print("nvdiffrast initialized successfully")
198
 
199
  print(f"torch.cuda.is_available():{torch.cuda.is_available()}")
 
193
  )
194
  )
195
  import nvdiffrast.torch as dr
196
+ print("import success")
197
+ dr.RasterizeCudaContext(device="cuda:0")
198
  print("nvdiffrast initialized successfully")
199
 
200
  print(f"torch.cuda.is_available():{torch.cuda.is_available()}")