StoryDiffusion icon indicating copy to clipboard operation
StoryDiffusion copied to clipboard

Update predict.py

Open yuxumin opened this issue 1 year ago • 1 comments

yuxumin avatar May 07 '24 08:05 yuxumin

The common issue, as seen in examples like the https://github.com/TencentARC/PhotoMaker/issues/55 or https://github.com/ZHO-ZHO-ZHO/ComfyUI-PhotoMaker-ZHO/issues/56, is caused by the presence of the [NC] keyword in the comic description. This keyword leads to a trigger word of length 0 when dealing with the class_token_index.

However, the assertion in the main code overlooks this issue (caused by a typo).

yuxumin avatar May 07 '24 08:05 yuxumin

@yuxumin Hi,[NC] is support now, could you update it?

Z-YuPeng avatar May 07 '24 19:05 Z-YuPeng

Already tried, now [NC] mode is well supported under photomaker, cool.

yuxumin avatar May 07 '24 23:05 yuxumin