Instructions to use stojchet/sft_small with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use stojchet/sft_small with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("deepseek-ai/deepseek-coder-1.3b-base") model = PeftModel.from_pretrained(base_model, "stojchet/sft_small") - Notebooks
- Google Colab
- Kaggle
| {"max_seq_length": 10, "language": "python", "dataset_size": 10, "epochs": 1, "per_device_train_batch_size": 1, "gradient_accumulation_steps": 1, "learning_rate": 1.41e-05, "lora_r": 64, "lora_alpha": 16, "lora_dropout": 0.1, "dataset_name": "stojchet/csn_java_python_subset", "base_model": "deepseek-ai/deepseek-coder-1.3b-base", "dataset_ref_field": "whole_func_string"} |