File size: 282 Bytes
74b17e0 |
1 2 3 4 5 6 7 8 9 10 |
from transformers import StableLmForCausalLM, AutoTokenizer
from . import register_llm
@register_llm('stablelm')
def return_phiclass():
def tokenizer_and_post_load(tokenizer):
return tokenizer
return (StableLmForCausalLM, (AutoTokenizer, tokenizer_and_post_load))
|