Wording in the code example

#2
by multimodalart HF staff - opened
No description provided.

The sampling needs to be currently wrapped in with torch.cuda.amp.autocast(dtype=dtype): for bfloat16 to work

Amazing, thanks!!!

dome272 changed pull request status to merged

Sign up or log in to comment