Views
No views yet
pip install transformers1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "ricardozhy/Qwen1.5-7B-poem"
4tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
5model = AutoModelForCausalLM.from_pretrained(
6 model_id,
7 torch_dtype="auto",
8 device_map="auto",
9 trust_remote_code=True
10)1from modelscope import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "njauzwh/Qwen1.5-7B-poem"
4tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
5model = AutoModelForCausalLM.from_pretrained(
6 model_id,
7 torch_dtype="auto",
8 device_map="auto",
9 trust_remote_code=True
10)