search

openai / consistency_models

Official repo for consistency models.
MIT License
6.17k stars 420 forks source link

Corrected RuntimeError in the Sample Code in README.md #43

Closed take2rohit closed 1 year ago

take2rohit commented 1 year ago

The previous version of code shows that it takes class_labels=145 as argument for pipe(). However this throws the below error.
"RuntimeError: Expected tensor for argument # 1 'indices' to have scalar type Long; but got torch.IntTensor instead (while checking arguments for embedding)"

We must pass a tensor with Long data type.

This resolves #44