3 ms·
Your `get_sigma_embeds(batches, sigma)` seems to not use its first input? Did you mean to broadcast sigma to shape (batches, 1)?
by thomasahle 3y ago
Your `get_sigma_embeds(batches, sigma)` seems to not use its first input?
Did you mean to broadcast sigma to shape (batches, 1)?
- ycy 3y agoMy intention was to omit the details of batching in the blog post for clarity of exposition, and I'll update the post accordingly. Sorry for the confusion, but you can see the full implementation here: https://github.com/yuanchenyang/smalldiffusion/blob/main/src/smalldiffusion/model.py https://github.com/yuanchenyang/smalldiffusion/blob/main/src...