metadata
license: llama2
AIMI FMs: A Collection of Foundation Models in Radiology
π Paper β’ π€ Hugging Face ⒠𧩠Github β’ πͺ Project
β¨ Latest News
- [01/20/2023]: Model released in Hugging Face.
π¬ Get Started
from transformers import AutoTokenizer
from transformers import AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("StanfordAIMI/RadLLaMA-7b", trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained("StanfordAIMI/RadLLaMA-7b")
prompt = "Hi"
conv = [{"from": "human", "value": prompt}]
input_ids = tokenizer.apply_chat_template(conv, add_generation_prompt=True, return_tensors="pt")
outputs = model.generate(input_ids)
response = tokenizer.decode(outputs[0])
print(response)
βοΈ Citation
@article{aimifms-2024,
title={},
author={},
journal={arXiv preprint arXiv:xxxx.xxxxx},
url={https://arxiv.org/abs/xxxx.xxxxx},
year={2024}
}