xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2025-12-10 05:14:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added Node Progress Bar

Browse Source
This commit is contained in:
Easymode 2025-02-20 01:02:47 +00:00 committed by GitHub
parent f5408c9493
commit 2b44651d83
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
hy3dgen/shapegen/bpt/model/model.py
View File

@ -18,6 +18,7 @@ from .miche_conditioner import PointConditioner
from functools import partial from functools import partial
from tqdm import tqdm from tqdm import tqdm
from .data_utils import discretize from .data_utils import discretize
from comfy.utils import ProgressBar
# helper functions # helper functions
@ -182,6 +183,10 @@ class MeshTransformer(Module):
cache = None cache = None
eos_iter = None eos_iter = None
# ✅ Initialize ComfyUI progress bar
pbar = ProgressBar(max_seq_len - curr_length)
# predict tokens auto-regressively # predict tokens auto-regressively
for i in tqdm(range(curr_length, max_seq_len), position=tqdm_position, for i in tqdm(range(curr_length, max_seq_len), position=tqdm_position,
desc=f'Process: {tqdm_position}', dynamic_ncols=True, leave=False): desc=f'Process: {tqdm_position}', dynamic_ncols=True, leave=False):
@ -207,6 +212,9 @@ class MeshTransformer(Module):
sample = torch.multinomial(probs, 1) sample = torch.multinomial(probs, 1)
codes, _ = pack([codes, sample], 'b *') codes, _ = pack([codes, sample], 'b *')
# ComfyUI progress bar
pbar.update(1)
# Check if all sequences have encountered EOS at least once # Check if all sequences have encountered EOS at least once
is_eos_codes = (codes == self.eos_token_id) is_eos_codes = (codes == self.eos_token_id)
if is_eos_codes.any(dim=-1).all(): if is_eos_codes.any(dim=-1).all():
@ -216,6 +224,9 @@ class MeshTransformer(Module):
# Once we've generated 20% more tokens than eos_iter, break out of the loop # Once we've generated 20% more tokens than eos_iter, break out of the loop
if codes.shape[-1] >= int(eos_iter * 1.2): if codes.shape[-1] >= int(eos_iter * 1.2):
break break
# Ensure progress bar reaches 100% when loop completes
pbar.complete()
# mask out to padding anything after the first eos # mask out to padding anything after the first eos