replicate / cog-sdxl

Stable Diffusion XL training and inference as a cog model
https://replicate.com/stability-ai/sdxl
Apache License 2.0
212 stars 106 forks source link

If a token_string contains commas then the trailing crashes #53

Open itsallmememe opened 6 months ago

itsallmememe commented 6 months ago

We're consistently getting crashes in the training script if we submit a comma in the token_string

Failing training model https://replicate.com/p/e76k0ze9a9rgm0cf1cftefcr0m?input=form

Traceback (most recent call last):
File "/usr/local/lib/python3.9/site-packages/cog/server/worker.py", line 217, in _predict
result = predict(**payload)
File "train.py", line 133, in train
n_tok = int(token.split(":")[1])
IndexError: list index out of range

Reproducible, crashes every time.