How to use chaitjo/all-atom-diffusion-transformer with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("chaitjo/all-atom-diffusion-transformer", dtype="auto")