Fitted 31da970
Fedor Krasnov commited on
How to use fkrasnov2/COLD2 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="fkrasnov2/COLD2") # Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("fkrasnov2/COLD2")
model = AutoModelForMaskedLM.from_pretrained("fkrasnov2/COLD2", device_map="auto")How to use fkrasnov2/COLD2 with Transformers.js:
// npm i @huggingface/transformers
import { pipeline } from '@huggingface/transformers';
// Allocate pipeline
const pipe = await pipeline('fill-mask', 'fkrasnov2/COLD2');