jorge-henao commited on
Commit
05e7f26
1 Parent(s): d571bc6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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)