RUCKBReasoning / codes

The source code of CodeS (SIGMOD 2024).
https://arxiv.org/abs/2402.16347
Apache License 2.0
139 stars 19 forks source link

codes-7b-merged load error #5

Closed ddingwang12 closed 8 months ago

ddingwang12 commented 8 months ago

when i load model

model = AutoModelForCausalLM.from_pretrained(model_name, device_map = "auto", torch_dtype = torch.float16)

some of pip package:

torch              1.13.1+cu117
torchaudio         0.13.1
torchvision        0.14.1
tokenizers         0.15.2
transformers       4.38.2
flash-attn         2.5.6

what is wrongwith me? and I found some related issues: https://github.com/oobabooga/text-generation-webui/issues/122#issuecomment-1560104217

lihaoyang-ruc commented 8 months ago

Hi!

As far as I know, the most common cause of this problem comes from missing some key files when downloading the model from Hugging Face.

Ensure you've downloaded all the files from codes-7b-merged and subsequently placed them into the /mnt/disk3/home/ls/project/text2sql-demo/model/seeklhy/codes directory.

ddingwang12 commented 8 months ago

ok,I redownloaded the model and now it works!but,I checked the SHA256 of the model I downloaded before and it is the same and Its location is no problem either。