Update README.md
Browse files
README.md
CHANGED
|
@@ -9,35 +9,41 @@ tags:
|
|
| 9 |
- transformers
|
| 10 |
- trl
|
| 11 |
- unsloth
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
-
# Model Card for Model ID
|
| 15 |
-
|
| 16 |
-
<!-- Provide a quick summary of what the model is/does. -->
|
| 17 |
|
|
|
|
| 18 |
|
|
|
|
| 19 |
|
| 20 |
## Model Details
|
| 21 |
|
| 22 |
### Model Description
|
| 23 |
|
| 24 |
-
|
|
|
|
|
|
|
| 25 |
|
| 26 |
|
| 27 |
|
| 28 |
-
- **Developed by:**
|
| 29 |
-
- **Funded by [optional]:**
|
| 30 |
-
- **Shared by [optional]:**
|
| 31 |
-
- **Model type:**
|
| 32 |
-
- **Language(s) (NLP):**
|
| 33 |
-
- **License:**
|
| 34 |
-
- **Finetuned from model [optional]:**
|
| 35 |
|
| 36 |
### Model Sources [optional]
|
| 37 |
|
| 38 |
<!-- Provide the basic links for the model. -->
|
| 39 |
|
| 40 |
-
- **Repository:**
|
| 41 |
- **Paper [optional]:** [More Information Needed]
|
| 42 |
- **Demo [optional]:** [More Information Needed]
|
| 43 |
|
|
@@ -47,45 +53,84 @@ tags:
|
|
| 47 |
|
| 48 |
### Direct Use
|
| 49 |
|
| 50 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
|
| 52 |
[More Information Needed]
|
| 53 |
|
| 54 |
### Downstream Use [optional]
|
| 55 |
|
| 56 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
|
| 58 |
[More Information Needed]
|
| 59 |
|
| 60 |
### Out-of-Scope Use
|
| 61 |
|
| 62 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
|
| 64 |
[More Information Needed]
|
| 65 |
|
| 66 |
## Bias, Risks, and Limitations
|
| 67 |
|
| 68 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 69 |
|
| 70 |
[More Information Needed]
|
| 71 |
|
| 72 |
### Recommendations
|
| 73 |
|
| 74 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 75 |
|
| 76 |
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
| 77 |
|
| 78 |
## How to Get Started with the Model
|
| 79 |
|
| 80 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 81 |
|
| 82 |
-
[More Information Needed]
|
| 83 |
|
| 84 |
## Training Details
|
| 85 |
|
| 86 |
### Training Data
|
| 87 |
|
| 88 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 89 |
|
| 90 |
[More Information Needed]
|
| 91 |
|
|
@@ -95,16 +140,32 @@ Use the code below to get started with the model.
|
|
| 95 |
|
| 96 |
#### Preprocessing [optional]
|
| 97 |
|
| 98 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 99 |
|
| 100 |
|
| 101 |
#### Training Hyperparameters
|
| 102 |
|
| 103 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 104 |
|
| 105 |
#### Speeds, Sizes, Times [optional]
|
| 106 |
|
| 107 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 108 |
|
| 109 |
[More Information Needed]
|
| 110 |
|
|
|
|
| 9 |
- transformers
|
| 10 |
- trl
|
| 11 |
- unsloth
|
| 12 |
+
- vcet
|
| 13 |
+
- domain-specific
|
| 14 |
+
license: apache-2.0
|
| 15 |
+
metrics:
|
| 16 |
+
- accuracy
|
| 17 |
---
|
| 18 |
|
|
|
|
|
|
|
|
|
|
| 19 |
|
| 20 |
+
# Model Card for gemma-270-it-vcet-lora
|
| 21 |
|
| 22 |
+
<!-- Provide a quick summary of what the model is/does. -->
|
| 23 |
|
| 24 |
## Model Details
|
| 25 |
|
| 26 |
### Model Description
|
| 27 |
|
| 28 |
+
This model is a domain-specific conversational AI fine-tuned on custom data related to VCET College, Madurai.
|
| 29 |
+
Built on top of unsloth/gemma-3-270m-it-unsloth-bnb-4bit, it uses LoRA and PEFT for efficient adaptation.
|
| 30 |
+
The model is designed to answer queries about campus life, academics, departments, events, and administrative processes at VCET.
|
| 31 |
|
| 32 |
|
| 33 |
|
| 34 |
+
- **Developed by:** SandeepCodez.
|
| 35 |
+
- **Funded by [optional]:** Self-funded.
|
| 36 |
+
- **Shared by [optional]:** SandeepCodez
|
| 37 |
+
- **Model type:** Causal Language Model (Text Generation).
|
| 38 |
+
- **Language(s) (NLP):** English (with contextual Tamil understanding).
|
| 39 |
+
- **License:** Apache 2.0.
|
| 40 |
+
- **Finetuned from model [optional]:** unsloth/gemma-3-270m-it-unsloth-bnb-4bit
|
| 41 |
|
| 42 |
### Model Sources [optional]
|
| 43 |
|
| 44 |
<!-- Provide the basic links for the model. -->
|
| 45 |
|
| 46 |
+
- **Repository:** https://github.com/SandeepCodez
|
| 47 |
- **Paper [optional]:** [More Information Needed]
|
| 48 |
- **Demo [optional]:** [More Information Needed]
|
| 49 |
|
|
|
|
| 53 |
|
| 54 |
### Direct Use
|
| 55 |
|
| 56 |
+
Answering VCET-related questions
|
| 57 |
+
|
| 58 |
+
Assisting students with academic and campus queries
|
| 59 |
+
|
| 60 |
+
Automating college FAQs
|
| 61 |
+
|
| 62 |
+
Supporting chatbot integration for VCET platforms
|
| 63 |
|
| 64 |
[More Information Needed]
|
| 65 |
|
| 66 |
### Downstream Use [optional]
|
| 67 |
|
| 68 |
+
Integration into college ERP systems
|
| 69 |
+
|
| 70 |
+
Enhancing virtual assistants for student support
|
| 71 |
+
|
| 72 |
+
Embedding in mobile apps or websites
|
| 73 |
|
| 74 |
[More Information Needed]
|
| 75 |
|
| 76 |
### Out-of-Scope Use
|
| 77 |
|
| 78 |
+
General-purpose text generation outside VCET context
|
| 79 |
+
|
| 80 |
+
Legal, medical, or financial advice
|
| 81 |
+
|
| 82 |
+
High-stakes decision-making without human oversight
|
| 83 |
|
| 84 |
[More Information Needed]
|
| 85 |
|
| 86 |
## Bias, Risks, and Limitations
|
| 87 |
|
| 88 |
+
May reflect institutional bias from VCET sources
|
| 89 |
+
|
| 90 |
+
Limited generalization outside VCET domain
|
| 91 |
+
|
| 92 |
+
Not suitable for sensitive or critical applications
|
| 93 |
|
| 94 |
[More Information Needed]
|
| 95 |
|
| 96 |
### Recommendations
|
| 97 |
|
| 98 |
+
Use in supervised environments
|
| 99 |
+
|
| 100 |
+
Periodic updates to dataset recommended
|
| 101 |
+
|
| 102 |
+
Human validation for factual accuracy advised
|
| 103 |
|
| 104 |
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
| 105 |
|
| 106 |
## How to Get Started with the Model
|
| 107 |
|
| 108 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 109 |
+
|
| 110 |
+
model_id = "SandeepCodez/gemma-270-it-vcet-lora"
|
| 111 |
+
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 112 |
+
model = AutoModelForCausalLM.from_pretrained(model_id)
|
| 113 |
+
|
| 114 |
+
inputs = tokenizer("What are the placement statistics for VCET Madurai?", return_tensors="pt")
|
| 115 |
+
outputs = model.generate(**inputs)
|
| 116 |
+
print(tokenizer.decode(outputs[0]))
|
| 117 |
|
|
|
|
| 118 |
|
| 119 |
## Training Details
|
| 120 |
|
| 121 |
### Training Data
|
| 122 |
|
| 123 |
+
Custom dataset created by the developer, including:
|
| 124 |
+
|
| 125 |
+
VCET brochures
|
| 126 |
+
|
| 127 |
+
Departmental documents
|
| 128 |
+
|
| 129 |
+
Student interviews
|
| 130 |
+
|
| 131 |
+
Campus FAQs
|
| 132 |
+
|
| 133 |
+
Event archives
|
| 134 |
|
| 135 |
[More Information Needed]
|
| 136 |
|
|
|
|
| 140 |
|
| 141 |
#### Preprocessing [optional]
|
| 142 |
|
| 143 |
+
Cleaned and structured into JSONL format
|
| 144 |
+
|
| 145 |
+
Tokenized using Gemma tokenizer
|
| 146 |
+
|
| 147 |
+
Filtered for relevance and clarity
|
| 148 |
|
| 149 |
|
| 150 |
#### Training Hyperparameters
|
| 151 |
|
| 152 |
+
Training regime: bf16 mixed precision
|
| 153 |
+
|
| 154 |
+
Epochs: 3
|
| 155 |
+
|
| 156 |
+
Batch Size: 16
|
| 157 |
+
|
| 158 |
+
Learning Rate: 2e-4
|
| 159 |
+
|
| 160 |
+
Frameworks: PEFT 0.17.1, TRL, Unsloth
|
| 161 |
|
| 162 |
#### Speeds, Sizes, Times [optional]
|
| 163 |
|
| 164 |
+
Training Time: ~3 hours
|
| 165 |
+
|
| 166 |
+
Dataset Size: ~10,000 samples
|
| 167 |
+
|
| 168 |
+
Model Size: 270M parameters
|
| 169 |
|
| 170 |
[More Information Needed]
|
| 171 |
|