Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2tok = AutoTokenizer.from_pretrained('TAUR-dev/M-R1_distilled_baseline_cd3args_only')
3model = AutoModelForCausalLM.from_pretrained('TAUR-dev/M-R1_distilled_baseline_cd3args_only', torch_dtype='auto')