- PromptlyCode inline completion model by PyTorch
- Tab automatically complete code and predict code use vscode inline completion
- Setup python env
conda create -n inline-completion-model python=3.11
conda activate inline-completion-model
poetry install- Input, Output Prediction
Input: "def func(x):"
Output Prediction: "return x * x"
Input: "for i in range(n):"
Target: "sum += i"
(['def func(x):', 'x = x + 1', 'return x'],
['x = x + 1', 'return x', 'print(func(5))'])
- Process Flowchart
Parse Python Files ─> Build Dataset ─> Train Model ─> Test Model
│ │ │ │
File Parsing Vocabulary Training Loop Predict
│ Creation ↓ ↓
Input/Target Pairs ─> DataLoader ─> Loss Optimization ─> Generate Output
