From 420c85895152e3156b24261b63bb6766257dac6f Mon Sep 17 00:00:00 2001 From: Perfect_7613 Date: Wed, 28 May 2025 23:17:36 +0530 Subject: [PATCH] Changed the delete trelis model to available one Removed the unvailable trelis model on huggingface to the model which is available on huggingface and works the same --- blend_gaussians.py | 6 +++--- run_pipeline.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/blend_gaussians.py b/blend_gaussians.py index 680b94d..377f445 100644 --- a/blend_gaussians.py +++ b/blend_gaussians.py @@ -520,8 +520,8 @@ def merge_gaussians( scene_vanilla = None scene = None - ss_decoder = models.from_pretrained("JeffreyXiang/TRELLIS-image-large/ckpts/ss_dec_conv3d_16l8_fp16").eval().cuda() - pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large") + ss_decoder = models.from_pretrained("jetx/trellis-image-large/ckpts/ss_dec_conv3d_16l8_fp16").eval().cuda() + pipeline = TrellisImageTo3DPipeline.from_pretrained("jetx/trellis-image-large") pipeline.cuda() rescaled_tiles = {} @@ -634,7 +634,7 @@ def merge_gaussians( UNCOND=False if stitch_slats: - pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large") + pipeline = TrellisImageTo3DPipeline.from_pretrained("jetx/trellis-image-large") pipeline.cuda() for tile in tqdm(grid_info, desc="Stitching slats"): x, y, = [tile[k] for k in ('x', 'y')] diff --git a/run_pipeline.py b/run_pipeline.py index 63867b8..558e01d 100644 --- a/run_pipeline.py +++ b/run_pipeline.py @@ -543,7 +543,7 @@ def worker(prefix, tile_dict, gradio_url, blender_path, gpu_queue, generated_gri def load_pipeline_in_thread(event, queue): # Load the pipeline in a separate thread to avoid blocking the main thread - thread_pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large") + thread_pipeline = TrellisImageTo3DPipeline.from_pretrained("jetx/trellis-image-large") thread_pipeline.to("cuda") queue.put(thread_pipeline) event.set()