Commit
·
05e7f26
1
Parent(s):
d571bc6
Update README.md
Browse files
README.md
CHANGED
@@ -69,7 +69,7 @@ import torch
|
|
69 |
from peft import PeftModel, PeftConfig
|
70 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
71 |
|
72 |
-
peft_model_id = "hackathon-somos-nlp-2023/baizemocracy-lora-7B-cfqa"
|
73 |
config = PeftConfig.from_pretrained(peft_model_id)
|
74 |
base_model = AutoModelForCausalLM.from_pretrained("decapoda-research/llama-7b-hf", return_dict=True, load_in_8bit=True, device_map='auto')
|
75 |
tokenizer = AutoTokenizer.from_pretrained(peft_model_id)
|
|
|
69 |
from peft import PeftModel, PeftConfig
|
70 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
71 |
|
72 |
+
peft_model_id = "hackathon-somos-nlp-2023/baizemocracy-lora-7B-cfqa-conv"
|
73 |
config = PeftConfig.from_pretrained(peft_model_id)
|
74 |
base_model = AutoModelForCausalLM.from_pretrained("decapoda-research/llama-7b-hf", return_dict=True, load_in_8bit=True, device_map='auto')
|
75 |
tokenizer = AutoTokenizer.from_pretrained(peft_model_id)
|