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 |
# 중간 메모리 정리
|