somosnlp
/

spanish_medica_llm

Text Generation

Generated from Trainer

Model card Files Files and versions Community

inoid commited on Apr 25, 2024

Commit

8d22e68

•

1 Parent(s): 6f943db

Update README.md

Files changed (1) hide show

README.md +3 -2

README.md CHANGED Viewed

@@ -108,8 +108,9 @@ de los resultados generados.
 Use the code below to get started with the model.
 ```
-[More Information Needed]
 ```

 Use the code below to get started with the model.
 ```
+HUB_MODEL_ID = '/somosnlp/spanish_medica_llm'
+config = PeftConfig.from_pretrained(HUB_MODEL_ID)
+model = AutoModelForCausalLM.from_pretrained(HUB_MODEL_ID, return_dict=True, load_in_8bit=True, device_map={"":0})
 ```