Duo Li

Issue with `use_fp16=True` Leading to Type Conversion Error in `unet.py`

1

When setting `use_fp16=False`, the code functions correctly. However, an issue arises with `use_fp16=True` due to an unexpected type conversion in [`unet.py`](https://github.com/openai/consistency_models/blob/main/cm/unet.py)(line435). The problem occurs at line 435, where the tensor...

I've been extensively experimenting with this incredible project and have found most implementations to function seamlessly. However, I've encountered several issues specifically related to iteration generation that I hope to...