* make it torch.compile comaptible * make the text encoder compatible too. * style
Attach a file by drag & drop or click to upload
