Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3repo_id = "sagnikM/qwen_qwen_step100_actor"
4model = AutoModelForCausalLM.from_pretrained(repo_id, torch_dtype="auto")
5tokenizer = AutoTokenizer.from_pretrained(repo_id)