How to use d0rj/diffusion-51M-base-v1-failed with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="d0rj/diffusion-51M-base-v1-failed", trust_remote_code=True)
# pip install -U transformers accelerate # Load model directly from transformers import AutoModelForMaskedLM model = AutoModelForMaskedLM.from_pretrained("d0rj/diffusion-51M-base-v1-failed", trust_remote_code=True, device_map="auto")
The community tab is the place to discuss and collaborate with the HF community!
