Downloads · 30 days
13
30% of all-time downloads
Santosh-Gupta/EncephalitisGPT
EncephalitisGPT is a text generation model from Santosh-Gupta. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as mit.
The training of model used Microsoft's bioGPT as a base model, and then finetuned in a dataset of over 370k encephalitis question and answers
Downloads · 30 days
13
30% of all-time downloads
All-time downloads
44
Public
Parameters
1.6B
6.3 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors6.3 GB · 100%
From the Hugging Face model README
The training of model used Microsoft's bioGPT as a base model, and then finetuned in a dataset of over 370k encephalitis question and answers
https://huggingface.co/datasets/Santosh-Gupta/EncephalitisQuestionAnswering
To use
from transformers import pipeline, AutoTokenizer, AutoModel
model = AutoModel.from_pretrained('Santosh-Gupta/EncephalitisGPT')
tokenizer = AutoTokenizer.from_pretrained('Santosh-Gupta/EncephalitisGPT')
generator = pipeline('text-generation', model=model, tokenizer=tokenizer, device=0)
generator("What are causes of autoimmune encephalitis", max_length=1000, num_return_sequences=1, do_sample=True)