Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -234,7 +234,7 @@ def image_to_3d(trial_id: str, seed: int, randomize_seed: bool, ss_guidance_stre
|
|
234 |
# 파이프라인을 GPU로 이동
|
235 |
g.trellis_pipeline.to('cuda')
|
236 |
|
237 |
-
#
|
238 |
outputs = g.trellis_pipeline.run(
|
239 |
image,
|
240 |
seed=seed,
|
@@ -242,14 +242,12 @@ def image_to_3d(trial_id: str, seed: int, randomize_seed: bool, ss_guidance_stre
|
|
242 |
preprocess_image=False,
|
243 |
sparse_structure_sampler_params={
|
244 |
"steps": min(ss_sampling_steps, 8),
|
245 |
-
"cfg_strength": ss_guidance_strength
|
246 |
-
"batch_size": 1
|
247 |
},
|
248 |
slat_sampler_params={
|
249 |
"steps": min(slat_sampling_steps, 8),
|
250 |
-
"cfg_strength": slat_guidance_strength
|
251 |
-
|
252 |
-
},
|
253 |
)
|
254 |
|
255 |
# 중간 메모리 정리
|
|
|
234 |
# 파이프라인을 GPU로 이동
|
235 |
g.trellis_pipeline.to('cuda')
|
236 |
|
237 |
+
# batch_size 제거하고 파라미터 수정
|
238 |
outputs = g.trellis_pipeline.run(
|
239 |
image,
|
240 |
seed=seed,
|
|
|
242 |
preprocess_image=False,
|
243 |
sparse_structure_sampler_params={
|
244 |
"steps": min(ss_sampling_steps, 8),
|
245 |
+
"cfg_strength": ss_guidance_strength
|
|
|
246 |
},
|
247 |
slat_sampler_params={
|
248 |
"steps": min(slat_sampling_steps, 8),
|
249 |
+
"cfg_strength": slat_guidance_strength
|
250 |
+
}
|
|
|
251 |
)
|
252 |
|
253 |
# 중간 메모리 정리
|