inoid commited on
Commit
9368020
1 Parent(s): 2c7b003

Update README_es.md

Browse files
Files changed (1) hide show
  1. README_es.md +6 -3
README_es.md CHANGED
@@ -113,9 +113,12 @@ de los resultados generados.
113
  Utilice el siguiente código para comenzar con el modelo.
114
 
115
  ```
116
- HUB_MODEL_ID = '/somosnlp/spanish_medica_llm'
117
- config = PeftConfig.from_pretrained(HUB_MODEL_ID)
118
- model = AutoModelForCausalLM.from_pretrained(HUB_MODEL_ID, return_dict=True, load_in_8bit=True, device_map={"":0})
 
 
 
119
 
120
  ```
121
 
 
113
  Utilice el siguiente código para comenzar con el modelo.
114
 
115
  ```
116
+ from peft import PeftModel, PeftConfig
117
+ from transformers import AutoModelForCausalLM
118
+
119
+ config = PeftConfig.from_pretrained("somosnlp/spanish_medica_llm")
120
+ model = AutoModelForCausalLM.from_pretrained("BioMistral/BioMistral-7B")
121
+ model = PeftModel.from_pretrained(model, "somosnlp/spanish_medica_llm")
122
 
123
  ```
124