In line 136 of the file train_spice.py, the training code:
model_kwargs = dict(x_start=prompts, cond=cond)
is not the same with the test code in line 181:
model_kwargs = dict(texts=prompts, cond=cond)
The x_start=prompts in line 136 should be texts=prompts.
In line 136 of the file train_spice.py, the training code:
model_kwargs = dict(x_start=prompts, cond=cond)is not the same with the test code in line 181:
model_kwargs = dict(texts=prompts, cond=cond)The
x_start=promptsin line 136 should betexts=prompts.