From d8002428b97b41558f70323dee47acf6f13cdfbb Mon Sep 17 00:00:00 2001 From: Javid Lakha <32059823+javidlakha@users.noreply.github.com> Date: Fri, 28 Feb 2025 16:44:02 -0500 Subject: [PATCH] Fix HuggingFace path for text completion quickstart --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 36f84f1..ff774e5 100644 --- a/README.md +++ b/README.md @@ -129,7 +129,7 @@ You can directly employ [Huggingface's Transformers](https://github.com/huggingf import torch from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig -model_name = "deepseek-ai/deepseek-ai/deepseek-moe-16b-base" +model_name = "deepseek-ai/deepseek-moe-16b-base" tokenizer = AutoTokenizer.from_pretrained(model_name) model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto") model.generation_config = GenerationConfig.from_pretrained(model_name)