Text Generation
Transformers
PyTorch
Safetensors
t5
text2text-generation
code
text-generation-inference
Instructions to use joernio/codetidal5 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use joernio/codetidal5 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="joernio/codetidal5")# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("joernio/codetidal5") model = AutoModelForSeq2SeqLM.from_pretrained("joernio/codetidal5") - Notebooks
- Google Colab
- Kaggle
- Local Apps
- vLLM
How to use joernio/codetidal5 with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "joernio/codetidal5" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "joernio/codetidal5", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/joernio/codetidal5
- SGLang
How to use joernio/codetidal5 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 "joernio/codetidal5" \ --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": "joernio/codetidal5", "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 "joernio/codetidal5" \ --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": "joernio/codetidal5", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - Docker Model Runner
How to use joernio/codetidal5 with Docker Model Runner:
docker model run hf.co/joernio/codetidal5
model description and paper/repo links
Browse files
README.md
CHANGED
|
@@ -8,4 +8,28 @@ library_name: transformers
|
|
| 8 |
pipeline_tag: text2text-generation
|
| 9 |
tags:
|
| 10 |
- code
|
| 11 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
pipeline_tag: text2text-generation
|
| 9 |
tags:
|
| 10 |
- code
|
| 11 |
+
---
|
| 12 |
+
|
| 13 |
+
# CodeTIDAL5
|
| 14 |
+
|
| 15 |
+
We present CodeTIDAL5, a model for type inference on untyped TypeScript / JavaScript!
|
| 16 |
+
The model was introduced as part of the paper
|
| 17 |
+
|
| 18 |
+
[_Learning Type Inference for Enhanced Dataflow Analysis_](https://davidbakereffendi.github.io/assets/pdf/preprint_6676_ESORICS23.pdf)
|
| 19 |
+
Lukas Seidel, Sedick David Baker Effendi, Xavier Pinho, Konrad Rieck, Brink van der Merwe and Fabian Yamaguchi
|
| 20 |
+
ESORICS 2023
|
| 21 |
+
|
| 22 |
+
From the abstract:
|
| 23 |
+
We propose CodeTIDAL5, a Transformer-based model trained to reliably
|
| 24 |
+
predict type annotations. For effective result retrieval and re-integration,
|
| 25 |
+
we extract usage slices from a program’s code property graph.
|
| 26 |
+
Comparing our approach against recent neural type inference systems, our
|
| 27 |
+
model outperforms the current state-of-the-art by 7.85% on the ManyTypes4TypeScript benchmark, achieving 71.27% accuracy overall.
|
| 28 |
+
|
| 29 |
+
## Intended Use
|
| 30 |
+
The model was designed for use with the code analysis platform [Joern](https://github.com/joernio/joern).
|
| 31 |
+
As part of the paper, we devise a system which seemlessly integrates type inference recommendations from the CodeTIDAL5 model in Joern's
|
| 32 |
+
Code Property Graphs (CPGs) for enriched context information, aiming at improved taint tracking and dataflow analysis.
|
| 33 |
+
|
| 34 |
+
An implementation of this approach can be found in the paper's artifact repository:
|
| 35 |
+
https://github.com/joernio/joernti-codetidal5
|