search

HVision-NKU / StoryDiffusion

Create Magic Story!
Apache License 2.0
5.45k stars 519 forks source link

Update predict.py #45

Closed yuxumin closed 2 months ago

yuxumin commented 2 months ago

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

Z-YuPeng commented 2 months ago

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

yuxumin commented 2 months ago

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