Instructions to use nferruz/ProtGPT2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use nferruz/ProtGPT2 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="nferruz/ProtGPT2")# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("nferruz/ProtGPT2") model = AutoModelForCausalLM.from_pretrained("nferruz/ProtGPT2") - Notebooks
- Google Colab
- Kaggle
- Local Apps
- vLLM
How to use nferruz/ProtGPT2 with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "nferruz/ProtGPT2" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "nferruz/ProtGPT2", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/nferruz/ProtGPT2
- SGLang
How to use nferruz/ProtGPT2 with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "nferruz/ProtGPT2" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "nferruz/ProtGPT2", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "nferruz/ProtGPT2" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "nferruz/ProtGPT2", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - Docker Model Runner
How to use nferruz/ProtGPT2 with Docker Model Runner:
docker model run hf.co/nferruz/ProtGPT2
Predicting end of sequence
Hello! I am trying to use the model to predict whether or not a protein sequence is complete (as opposed to erroneously truncated). It seems as though ProtGPT2 model tends to extend already-complete protein sequences (i.e. it would rarely predict 0 as the next token). I wanted to double-check these results using the API on the browser, which led to the following observation: I noticed that it never predicts and <|end of text|> token -- it simply spits out a warning saying "no text generated." Does this mean that the model has concluded the input sequence is complete?
Thank you!
Hi!
The model will terminate sequences producing an <|endoftext|> token, but it won't appear during generation because special tokens are not displayed unless you choose to. Bear in mind that this model tends to generate sequences a bit longer than natural on average (it likes to speak a lot), so it may continue sequences that are already complete. Other models we've trained later do not show this behavior.
On a different note, what do you mean by the API in the browser? Do you mean here in HF? If so, I would not rely on the generation because it's automatically produced by HF without using the most optimal generation parameters. Also, it does not filter by perplexity. In any case, the HF API would not show the special character <|endoftext|>, so I guess the behaviour would be 'no text generated' as you said (but with no experience generating from this browser, I'd do it locally instead :)).
Hope this helps,
Noelia