yilong2001

Results 3 comments of yilong2001

用这种方式加载,也是一样的问题: ```python model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True, device_map='auto', torch_dtype=torch.bfloat16) model = PeftModel.from_pretrained(model, peft_model_id) model = model.eval() ```

如果这样加载(先做一次 eval): ```python tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True) model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True, device_map='auto', torch_dtype=torch.bfloat16) model = model.eval() model = PeftModel.from_pretrained(model, peft_model_id) ``` 在这一步会出现如下问题: ```bash ValueError: We need an `offload_dir` to dispatch...

> @okpatil4u implemented classification here #11 I don't know if they ever made the changes public Here is the bert classification model with ggml: https://github.com/yilong2001/berts.cpp