Spaces:
21amer
/
Runtime error

HongFangzhou commited on
Commit
b9cc090
1 Parent(s): c92026e

update assets

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -115,6 +115,7 @@ opt.save = GRADIO_SAVE_PATH_MESH
115
  opt.prompt = ''
116
  opt.text_dir = True
117
  opt.front_dir = '+z'
 
118
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
119
  gui = GUI(opt)
120
  ###################################### INIT STAGE 2 #########################################
@@ -140,7 +141,7 @@ def add_text(rgb, caption):
140
 
141
  def marching_cube(b, text, global_info):
142
  # prepare volumn for marching cube
143
- res = 128
144
  assert 'decode_res' in global_info
145
  decode_res = global_info['decode_res']
146
  c_list = torch.linspace(-1.2, 1.2, steps=res)
 
115
  opt.prompt = ''
116
  opt.text_dir = True
117
  opt.front_dir = '+z'
118
+ opt.force_cuda_rast = True
119
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
120
  gui = GUI(opt)
121
  ###################################### INIT STAGE 2 #########################################
 
141
 
142
  def marching_cube(b, text, global_info):
143
  # prepare volumn for marching cube
144
+ res = 64
145
  assert 'decode_res' in global_info
146
  decode_res = global_info['decode_res']
147
  c_list = torch.linspace(-1.2, 1.2, steps=res)