AI4Finance-Foundation / AI4Finance-Foundation/FinGPT

Error when trying to run with a Quantized base model

Đang mở
#138 1 bình luận 0 reaction 0 người được giao Xem trên GitHub
Ngôn ngữ chính
Jupyter Notebook
Star
21.2k
Fork
3k
Merge trung bình
1 giờ 55 phút
Pull request đã merge (30 ngày)
5

Mô tả

Hello. I have been trying to run the multi task llama7b models with Bloke's llama 7b GPTQ(https://huggingface.co/TheBloke/Llama-2-7B-GPTQ) as the base.

```
def load_model(base_model, peft_model, from_remote=True):

model_name = parse_model_name(base_model, from_remote)
# model = AutoModelForCausalLM.from_pretrained(
# model_name, trust_remote_code=True,
# device_map="auto",
# )
model_name_or_path = "TheBloke/Llama-2-7b-Chat-GPTQ"
model = AutoGPTQForCausalLM.from_quantized(model_name_or_path,
model_basename="model",
use_safetensors=True,
trust_remote_code=True,
device="cuda:0",
use_triton="False")
model.model_parallel = True

tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)

model = PeftModel.from_pretrained(model, peft_model)
model = model.eval()
return model, tokenizer
```
While running the same in google colab, I get the error when trying to load PEFT from pre trained:
```
ValueError: Target modules ['q_proj', 'k_proj', 'v_proj'] not found in the base model. Please check the target modules and try again.
```

After a bit of searching, It says I'll have to re train the PEFT model by using a different config. Is there anything I can do? (other than training)

For debugging purposes, value of 'model' before PEFT is used:
```
LlamaGPTQForCausalLM(
(model): LlamaForCausalLM(
(model): LlamaModel(
(embed_tokens): Embedding(32000, 4096, padding_idx=0)
(layers): ModuleList(
(0-31): 32 x LlamaDecoderLayer(
(self_attn): FusedLlamaAttentionForQuantizedModel(
(qkv_proj): GeneralQuantLinear(in_features=4096, out_features=12288, bias=True)
(o_proj): GeneralQuantLinear(in_features=4096, out_features=4096, bias=True)
(rotary_emb): LlamaRotaryEmbedding()
)
(mlp): FusedLlamaMLPForQuantizedModel(
(gate_proj): GeneralQuantLinear(in_features=4096, out_features=11008, bias=True)
(up_proj): GeneralQuantLinear(in_features=4096, out_features=11008, bias=True)
(down_proj): GeneralQuantLinear(in_features=11008, out_features=4096, bias=True)
)
(input_layernorm): LlamaRMSNorm()
(post_attention_layernorm): LlamaRMSNorm()
)
)
(norm): LlamaRMSNorm()
)
(lm_head): Linear(in_features=4096, out_features=32000, bias=False)
)
)
```

Hướng dẫn đóng góp

Mở hướng dẫn đóng góp

Đánh giá

Issue này chưa được đánh giá.

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.