theprint commited on
Commit
dcb95bd
·
verified ·
1 Parent(s): 62f81e2

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,145 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: google/gemma-3-1b-it
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ language: en
6
+ license: apache-2.0
7
+ tags:
8
+ - lora
9
+ - sft
10
+ - transformers
11
+ - trl
12
+ - unsloth
13
+ - fine-tuned
14
+ datasets:
15
+ - theprint/titles-n-tags-alpaca
16
+ ---
17
+ # TiTan-Gemma3-1B
18
+
19
+ A fine-tuned Gemma 3 1B model, fine tuned for generating conversation titles and tags.
20
+
21
+ ## Model Details
22
+
23
+ This model is a fine-tuned version of google/gemma-3-1b-it using the Unsloth framework with LoRA (Low-Rank Adaptation) for efficient training.
24
+
25
+ - **Developed by:** theprint
26
+ - **Model type:** Causal Language Model (Fine-tuned with LoRA)
27
+ - **Language:** en
28
+ - **License:** apache-2.0
29
+ - **Base model:** google/gemma-3-1b-it
30
+ - **Fine-tuning method:** LoRA with rank 128
31
+
32
+ ## Intended Use
33
+
34
+ Title and tag generation.
35
+
36
+ ## Training Details
37
+
38
+ ### Training Data
39
+
40
+ The titles-n-tags set was specifically created for finetuning models on titling and tagging.
41
+
42
+ - **Dataset:** theprint/titles-n-tags-alpaca
43
+ - **Format:** alpaca
44
+
45
+ ### Training Procedure
46
+
47
+ - **Training epochs:** 2
48
+ - **LoRA rank:** 128
49
+ - **Learning rate:** 0.0001
50
+ - **Batch size:** 6
51
+ - **Framework:** Unsloth + transformers + PEFT
52
+ - **Hardware:** NVIDIA RTX 5090
53
+
54
+ ## Usage
55
+
56
+ ```python
57
+ from unsloth import FastLanguageModel
58
+ import torch
59
+
60
+ # Load model and tokenizer
61
+ model, tokenizer = FastLanguageModel.from_pretrained(
62
+ model_name="theprint/TiTan-Gemma3-1B",
63
+ max_seq_length=4096,
64
+ dtype=None,
65
+ load_in_4bit=True,
66
+ )
67
+
68
+ # Enable inference mode
69
+ FastLanguageModel.for_inference(model)
70
+
71
+ # Example usage
72
+ inputs = tokenizer(["Your prompt here"], return_tensors="pt")
73
+ outputs = model.generate(**inputs, max_new_tokens=256, temperature=0.7)
74
+ response = tokenizer.decode(outputs[0], skip_special_tokens=True)
75
+ print(response)
76
+ ```
77
+
78
+ ### Alternative Usage (Standard Transformers)
79
+
80
+ ```python
81
+ from transformers import AutoModelForCausalLM, AutoTokenizer
82
+ import torch
83
+
84
+ model = AutoModelForCausalLM.from_pretrained(
85
+ "theprint/TiTan-Gemma3-1B",
86
+ torch_dtype=torch.float16,
87
+ device_map="auto"
88
+ )
89
+ tokenizer = AutoTokenizer.from_pretrained("theprint/TiTan-Gemma3-1B")
90
+
91
+ # Example usage
92
+ messages = [
93
+ {"role": "system", "content": "You are a helpful assistant."},
94
+ {"role": "user", "content": "Your question here"}
95
+ ]
96
+
97
+ inputs = tokenizer.apply_chat_template(messages, return_tensors="pt", add_generation_prompt=True)
98
+ outputs = model.generate(inputs, max_new_tokens=256, temperature=0.7, do_sample=True)
99
+ response = tokenizer.decode(outputs[0][inputs.shape[-1]:], skip_special_tokens=True)
100
+ print(response)
101
+ ```
102
+ ## GGUF Quantized Versions
103
+
104
+ Quantized GGUF versions are available in the `gguf/` directory for use with llama.cpp:
105
+
106
+ - `TiTan-Gemma3-1B-f16.gguf` (2489.6 MB) - 16-bit float (original precision, largest file)
107
+ - `TiTan-Gemma3-1B-q3_k_m.gguf` (850.9 MB) - 3-bit quantization (medium quality)
108
+ - `TiTan-Gemma3-1B-q4_k_m.gguf` (966.7 MB) - 4-bit quantization (medium, recommended for most use cases)
109
+ - `TiTan-Gemma3-1B-q5_k_m.gguf` (1027.9 MB) - 5-bit quantization (medium, good quality)
110
+ - `TiTan-Gemma3-1B-q6_k.gguf` (1270.9 MB) - 6-bit quantization (high quality)
111
+ - `TiTan-Gemma3-1B-q8_0.gguf` (1325.8 MB) - 8-bit quantization (very high quality)
112
+
113
+ ### Using with llama.cpp
114
+
115
+ ```bash
116
+ # Download a quantized version (q4_k_m recommended for most use cases)
117
+ wget https://huggingface.co/theprint/TiTan-Gemma3-1B/resolve/main/gguf/TiTan-Gemma3-1B-q4_k_m.gguf
118
+
119
+ # Run with llama.cpp
120
+ ./llama.cpp/main -m TiTan-Gemma3-1B-q4_k_m.gguf -p "Your prompt here" -n 256
121
+ ```
122
+ ## Limitations
123
+
124
+ May provide incorrect information.
125
+
126
+ ## Citation
127
+
128
+ If you use this model, please cite:
129
+
130
+ ```bibtex
131
+ @misc{titan_gemma3_1b,
132
+ title={TiTan-Gemma3-1B: Fine-tuned google/gemma-3-1b-it},
133
+ author={theprint},
134
+ year={2025},
135
+ publisher={Hugging Face},
136
+ url={https://huggingface.co/theprint/TiTan-Gemma3-1B}
137
+ }
138
+ ```
139
+
140
+ ## Acknowledgments
141
+
142
+ - Base model: [google/gemma-3-1b-it](https://huggingface.co/google/gemma-3-1b-it)
143
+ - Training dataset: [theprint/titles-n-tags-alpaca](https://huggingface.co/datasets/theprint/titles-n-tags-alpaca)
144
+ - Fine-tuning framework: [Unsloth](https://github.com/unslothai/unsloth)
145
+ - Quantization: [llama.cpp](https://github.com/ggerganov/llama.cpp)
added_tokens.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "<image_soft_token>": 262144
3
+ }
chat_template.jinja ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {{ bos_token }}
2
+ {%- if messages[0]['role'] == 'system' -%}
3
+ {%- if messages[0]['content'] is string -%}
4
+ {%- set first_user_prefix = messages[0]['content'] + '
5
+
6
+ ' -%}
7
+ {%- else -%}
8
+ {%- set first_user_prefix = messages[0]['content'][0]['text'] + '
9
+
10
+ ' -%}
11
+ {%- endif -%}
12
+ {%- set loop_messages = messages[1:] -%}
13
+ {%- else -%}
14
+ {%- set first_user_prefix = "" -%}
15
+ {%- set loop_messages = messages -%}
16
+ {%- endif -%}
17
+ {%- for message in loop_messages -%}
18
+ {%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
19
+ {{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
20
+ {%- endif -%}
21
+ {%- if (message['role'] == 'assistant') -%}
22
+ {%- set role = "model" -%}
23
+ {%- else -%}
24
+ {%- set role = message['role'] -%}
25
+ {%- endif -%}
26
+ {{ '<start_of_turn>' + role + '
27
+ ' + (first_user_prefix if loop.first else "") }}
28
+ {%- if message['content'] is string -%}
29
+ {{ message['content'] | trim }}
30
+ {%- elif message['content'] is iterable -%}
31
+ {%- for item in message['content'] -%}
32
+ {%- if item['type'] == 'image' -%}
33
+ {{ '<start_of_image>' }}
34
+ {%- elif item['type'] == 'text' -%}
35
+ {{ item['text'] | trim }}
36
+ {%- endif -%}
37
+ {%- endfor -%}
38
+ {%- else -%}
39
+ {{ raise_exception("Invalid content type") }}
40
+ {%- endif -%}
41
+ {{ '<end_of_turn>
42
+ ' }}
43
+ {%- endfor -%}
44
+ {%- if add_generation_prompt -%}
45
+ {{'<start_of_turn>model
46
+ '}}
47
+ {%- endif -%}
config.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_sliding_window_pattern": 6,
3
+ "architectures": [
4
+ "Gemma3ForCausalLM"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "attn_logit_softcapping": null,
9
+ "bos_token_id": 2,
10
+ "cache_implementation": "hybrid",
11
+ "eos_token_id": [
12
+ 1,
13
+ 106
14
+ ],
15
+ "final_logit_softcapping": null,
16
+ "head_dim": 256,
17
+ "hidden_activation": "gelu_pytorch_tanh",
18
+ "hidden_size": 1152,
19
+ "initializer_range": 0.02,
20
+ "intermediate_size": 6912,
21
+ "layer_types": [
22
+ "sliding_attention",
23
+ "sliding_attention",
24
+ "sliding_attention",
25
+ "sliding_attention",
26
+ "sliding_attention",
27
+ "full_attention",
28
+ "sliding_attention",
29
+ "sliding_attention",
30
+ "sliding_attention",
31
+ "sliding_attention",
32
+ "sliding_attention",
33
+ "full_attention",
34
+ "sliding_attention",
35
+ "sliding_attention",
36
+ "sliding_attention",
37
+ "sliding_attention",
38
+ "sliding_attention",
39
+ "full_attention",
40
+ "sliding_attention",
41
+ "sliding_attention",
42
+ "sliding_attention",
43
+ "sliding_attention",
44
+ "sliding_attention",
45
+ "full_attention",
46
+ "sliding_attention",
47
+ "sliding_attention"
48
+ ],
49
+ "max_position_embeddings": 32768,
50
+ "model_type": "gemma3_text",
51
+ "num_attention_heads": 4,
52
+ "num_hidden_layers": 26,
53
+ "num_key_value_heads": 1,
54
+ "pad_token_id": 0,
55
+ "query_pre_attn_scalar": 256,
56
+ "rms_norm_eps": 1e-06,
57
+ "rope_local_base_freq": 10000,
58
+ "rope_scaling": null,
59
+ "rope_theta": 1000000,
60
+ "sliding_window": 512,
61
+ "torch_dtype": "float16",
62
+ "transformers_version": "4.53.2",
63
+ "use_cache": true,
64
+ "vocab_size": 262144
65
+ }
generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 2,
3
+ "cache_implementation": "hybrid",
4
+ "do_sample": true,
5
+ "eos_token_id": [
6
+ 1,
7
+ 106
8
+ ],
9
+ "pad_token_id": 0,
10
+ "top_k": 64,
11
+ "top_p": 0.95,
12
+ "transformers_version": "4.53.2"
13
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c3fa9fe3043cf0deb7609df12e203ad1b0aab512fdd25dc5a4ecc276f02e5a92
3
+ size 1999887347
special_tokens_map.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "boi_token": "<start_of_image>",
3
+ "bos_token": {
4
+ "content": "<bos>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ "eoi_token": "<end_of_image>",
11
+ "eos_token": {
12
+ "content": "<eos>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false
17
+ },
18
+ "image_token": "<image_soft_token>",
19
+ "pad_token": {
20
+ "content": "<pad>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false
25
+ },
26
+ "unk_token": {
27
+ "content": "<unk>",
28
+ "lstrip": false,
29
+ "normalized": false,
30
+ "rstrip": false,
31
+ "single_word": false
32
+ }
33
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4667f2089529e8e7657cfb6d1c19910ae71ff5f28aa7ab2ff2763330affad795
3
+ size 33384568
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1299c11d7cf632ef3b4e11937501358ada021bbdf7c47638d13c0ee982f2e79c
3
+ size 4689074
tokenizer_config.json ADDED
The diff for this file is too large to render. See raw diff