Spaces:
Sleeping
Sleeping
FrozenBurning
commited on
Commit
·
ef4bc82
1
Parent(s):
692682d
fix app.py
Browse files
app.py
CHANGED
@@ -141,7 +141,7 @@ def process(input_image, input_num_steps=25, input_seed=42, input_cfg=6.0):
|
|
141 |
# exporting GLB mesh
|
142 |
denoise_param_path = os.path.join(config.output_dir, 'denoised.pt')
|
143 |
primx_ckpt_weight = torch.load(denoise_param_path, map_location='cpu')['model_state_dict']
|
144 |
-
model_primx.load_state_dict(
|
145 |
model_primx.to(device)
|
146 |
model_primx.eval()
|
147 |
with torch.no_grad():
|
|
|
141 |
# exporting GLB mesh
|
142 |
denoise_param_path = os.path.join(config.output_dir, 'denoised.pt')
|
143 |
primx_ckpt_weight = torch.load(denoise_param_path, map_location='cpu')['model_state_dict']
|
144 |
+
model_primx.load_state_dict(primx_ckpt_weight)
|
145 |
model_primx.to(device)
|
146 |
model_primx.eval()
|
147 |
with torch.no_grad():
|