# Load model directly
from transformers import AutoModelForMaskedLM
model = AutoModelForMaskedLM.from_pretrained("monsoon-nlp/dna-blockdiff-2", trust_remote_code=True, dtype="auto")Quick Links
Test version of https://huggingface.co/monsoon-nlp/dna-blockdiff using the custom bd3lms EmbeddingLayer instead of the torch.nn.EmbeddingLayer
- Does not seem to stop generating
- Does not fix issue with attention on inference
- Does not fix issue with torch.where in training
- Downloads last month
- 8
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="monsoon-nlp/dna-blockdiff-2", trust_remote_code=True)