Downloads · 30 days
10
1% of all-time downloads
OPT-350m-COVID-Finetune
Kigo/OPT-350m-COVID-Finetune
OPT-350m-COVID-Finetune is a text generation model from Kigo. Use it when you need the model to write or continue text. It is set up for transformers.
All-time downloads
806
Public
Repo size
2.6 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin1.3 GB · 100%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- opt
- Access
- Public
- Created
- Sep 5, 2022
- Updated
- Sep 5, 2022
- SHA
- c5c9ef5f
Try a prompt
Model card
From the Hugging Face model README
Best Generations with
from transformers import OPTForCausalLM
from transformers import GPT2Tokenizer
model = OPTForCausalLM.from_pretrained("Kigo/OPT-350m-COVID-Finetune", from_tf=True)
tokenizer = GPT2Tokenizer.from_pretrained("facebook/opt-350m")
inputs = tokenizer("Covid-19 is Positive, 42.237% of Lungs show GGO, Lower Left Lobe was most affected, Upper Left Lobe was least affected, yes Vascular dilatation, found consolidation. The patient is in critical condition. \n\n\n ", return_tensors="pt")
generate_ids = model.generate(inputs.input_ids,
do_sample=True,
max_new_tokens=1000,
)
completion = tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
print(completion)
- Task
- Text Generation
- Library
- transformers
- Type
- opt
- Created
- Sep 5, 2022
- Updated
- Sep 5, 2022