Views
No views yet
LatexForCausalLM, model_type=latex)1import torch
2from transformers import AutoModelForCausalLM, AutoConfig
3
4# registers custom classes
5repo = "MagistrTheOne/NULLXES-L-TEX-100M-Stage0a-v0.1"
6config = AutoConfig.from_pretrained(repo, trust_remote_code=True)
7model = AutoModelForCausalLM.from_pretrained(
8 repo, trust_remote_code=True, torch_dtype=torch.bfloat16
9)tokenizer.model, special_tokens.json). Prefer loading via the research package LatexTokenizer from the NULLXES-LÆTEX GitHub when doing local QA.