From 259693af7b014ef8ae374b5ad0f0aa21ba1f0370 Mon Sep 17 00:00:00 2001 From: platformofai Date: Mon, 21 Apr 2025 09:45:16 +0000 Subject: [PATCH] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index baeb4d5..311122c 100644 --- a/README.md +++ b/README.md @@ -45,11 +45,11 @@ license: apache-2.0 from modelscope import AutoModelForCausalLM, AutoTokenizer model = AutoModelForCausalLM.from_pretrained( - "PAI/DistilQwen2.5-DS3-0324-32B", + "PAI/DistilQwen2.5-DS3-0324-7B", torch_dtype="auto", device_map="auto" ) -tokenizer = AutoTokenizer.from_pretrained("PAI/DistilQwen2.5-DS3-0324-32B") +tokenizer = AutoTokenizer.from_pretrained("PAI/DistilQwen2.5-DS3-0324-7B") prompt = "Give me a short introduction to large language model." messages=[