HongFangzhou
commited on
Commit
•
b9cc090
1
Parent(s):
c92026e
update assets
Browse files
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 =
|
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)
|