Harish2002 commited on
Commit
7e22977
·
verified ·
1 Parent(s): f351e7c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +26 -3
README.md CHANGED
@@ -1,3 +1,26 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ ---
4
+ ## 🧩 Base Model
5
+
6
+ - **Base**: [TinyLlama/TinyLlama-1.1B-Chat-v1.0](https://huggingface.co/TinyLlama/TinyLlama-1.1B-Chat-v1.0)
7
+ - **Fine-tuned using**: [LoRA (Low-Rank Adaptation)](https://arxiv.org/abs/2106.09685)
8
+ - **Frameworks**: Hugging Face `transformers`, `peft`, `datasets`, `accelerate`
9
+
10
+ ## 🗂️ Dataset
11
+
12
+ - Custom dataset of 150+ Q&A pairs on CLI topics (git, bash, grep, tar, venv)
13
+ - Stored in `cli_questions.json`
14
+ - Tokenized and saved in `tokenized_dataset/`
15
+
16
+ ## 🏗️ Fine-Tuning
17
+
18
+ ```python
19
+ base_model = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
20
+ lora_config = LoraConfig(
21
+ r=16,
22
+ lora_alpha=32,
23
+ lora_dropout=0.1,
24
+ bias="none",
25
+ task_type="CAUSAL_LM"
26
+ )