RuntimeError when the number of prompts from a file causes a partial sample batch

[ikarth]

Using the --from-file option and supplying a list of prompts that has a length that is not a multiple of n_samples causes the following error:

RuntimeError: einsum(): operands do not broadcast with remapped shapes [original->remapped]: [48, 4096, 40]->[48, 4096, 1, 40] [40, 77, 40]->[40, 1, 77, 40]

I'm on Windows 10, though I doubt it matters for this one.

[ikarth]

Ended up writing a pull request for how I fixed it in my local copy: #13

Which works for my purposes and fixes the error, but is not the only way to resolve it, so feel free to use or disregard as you see fit.