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()