Commit 4127f4af authored by AUTOMATIC's avatar AUTOMATIC
Browse files

fix program breaking on empty prompt

parent 2f2d356e

modules/prompt_parser.py

+3 −0
Original line number Diff line number Diff line
@@ -212,4 +212,7 @@ def parse_prompt_attention(text): 
    for pos in square_brackets:

        multiply_range(pos, square_bracket_multiplier)



    if len(res) == 0:

        res = [["", 1.0]]



    return res

CRA Git | Maintained and supported by SUSTech CRA and CCSE