aiqtech commited on
Commit
7d57d13
·
verified ·
1 Parent(s): 0769ffa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
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
- "batch_size": 1
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
  # 중간 메모리 정리