vamazing commited on
Commit
6f19c52
·
verified ·
1 Parent(s): 1b75095

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +36 -3
README.md CHANGED
@@ -16,6 +16,7 @@ library_name: transformers
16
  # Koa AI v2 (`vamazing/Koa-AI-v2-code-9B`)
17
 
18
  **Koa AI v2** is an advanced, instruction-tuned language model engineered for agentic workflows, complex code synthesis, multi-turn tool interaction, and step-by-step technical reasoning.
 
19
 
20
  This repository provides both **16-bit merged weights (`.safetensors`)** and **quantized local binaries (`.gguf`)** exported directly from **`checkpoint-270`** (optimal loss: `0.5614`).
21
 
@@ -26,7 +27,7 @@ This repository provides both **16-bit merged weights (`.safetensors`)** and **q
26
  | Feature | Specification |
27
  | :--- | :--- |
28
  | **Model Name** | Koa AI v2 (Code) |
29
- | **Base Architecture** | Qwen 2.5 9B |
30
  | **Parameters** | 9 Billion |
31
  | **Precision Formats** | 16-bit Merged (`bf16`) & GGUF (`Q4_K_M` / `Q8_0`) |
32
  | **Context Length** | 32,768 tokens native (Fine-tuned at 2,048 sequence cap) |
@@ -34,9 +35,10 @@ This repository provides both **16-bit merged weights (`.safetensors`)** and **q
34
  | **Target Modules** | `q_proj`, `k_proj`, `v_proj`, `o_proj`, `gate_proj`, `up_proj`, `down_proj` |
35
  | **Primary Frameworks** | Unsloth, PyTorch, Hugging Face Transformers, `llama.cpp` |
36
 
 
37
  ---
38
 
39
- ## Modalities & Capabilities
40
 
41
  ### Supported Modalities
42
  * **Text Input → Text/Code Output**: Structured reasoning, code synthesis, documentation, and agentic trajectory logging.
@@ -44,9 +46,10 @@ This repository provides both **16-bit merged weights (`.safetensors`)** and **q
44
 
45
  ### Core Capabilities
46
  * **Agentic Coding & Execution**: Fine-tuned on agentic interaction traces to analyze system states, execute terminal commands, write code, and autonomously debug execution errors.
47
- * **Qwen 2.5 9B Foundation**: Leverages deep multi-step problem solving across complex multi-file codebases and algorithm challenges.
48
  * **Structured Reasoning**: Native support for deep logic, architectural planning, and structured chain-of-thought processing.
49
 
 
50
  ---
51
 
52
  ## Quickstart
@@ -55,6 +58,36 @@ This repository provides both **16-bit merged weights (`.safetensors`)** and **q
55
 
56
  ```bash
57
  pip install transformers torch accelerate unsloth
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
58
 
59
  This qwen3_5 model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
60
 
 
16
  # Koa AI v2 (`vamazing/Koa-AI-v2-code-9B`)
17
 
18
  **Koa AI v2** is an advanced, instruction-tuned language model engineered for agentic workflows, complex code synthesis, multi-turn tool interaction, and step-by-step technical reasoning.
19
+ It is a fine-tuned 9B parameter language model built on the Qwen 3.5 9B architecture. It is optimized for lightweight text generation and coding tasks.
20
 
21
  This repository provides both **16-bit merged weights (`.safetensors`)** and **quantized local binaries (`.gguf`)** exported directly from **`checkpoint-270`** (optimal loss: `0.5614`).
22
 
 
27
  | Feature | Specification |
28
  | :--- | :--- |
29
  | **Model Name** | Koa AI v2 (Code) |
30
+ | **Base Architecture** | Qwen 3.5 9B |
31
  | **Parameters** | 9 Billion |
32
  | **Precision Formats** | 16-bit Merged (`bf16`) & GGUF (`Q4_K_M` / `Q8_0`) |
33
  | **Context Length** | 32,768 tokens native (Fine-tuned at 2,048 sequence cap) |
 
35
  | **Target Modules** | `q_proj`, `k_proj`, `v_proj`, `o_proj`, `gate_proj`, `up_proj`, `down_proj` |
36
  | **Primary Frameworks** | Unsloth, PyTorch, Hugging Face Transformers, `llama.cpp` |
37
 
38
+
39
  ---
40
 
41
+ ## 📡 Modalities & Capabilities
42
 
43
  ### Supported Modalities
44
  * **Text Input → Text/Code Output**: Structured reasoning, code synthesis, documentation, and agentic trajectory logging.
 
46
 
47
  ### Core Capabilities
48
  * **Agentic Coding & Execution**: Fine-tuned on agentic interaction traces to analyze system states, execute terminal commands, write code, and autonomously debug execution errors.
49
+ * **Qwen 3.5 9B Foundation**: Leverages deep multi-step problem solving across complex multi-file codebases and algorithm challenges.
50
  * **Structured Reasoning**: Native support for deep logic, architectural planning, and structured chain-of-thought processing.
51
 
52
+
53
  ---
54
 
55
  ## Quickstart
 
58
 
59
  ```bash
60
  pip install transformers torch accelerate unsloth
61
+ ```
62
+
63
+ ```python
64
+ from unsloth import FastLanguageModel
65
+
66
+ # 1. Load the model and tokenizer
67
+ model, tokenizer = FastLanguageModel.from_pretrained(
68
+ model_name = "your-username/Koa-AI-v1",
69
+ max_seq_length = 2048,
70
+ load_in_4bit = True,
71
+ )
72
+ FastLanguageModel.for_inference(model)
73
+
74
+ # 2. Define prompt using ChatML template
75
+ messages = [
76
+ {"role": "system", "content": "You are Koa AI v1, an expert coding agent."},
77
+ {"role": "user", "content": "Write a Python script to monitor GPU VRAM usage."},
78
+ ]
79
+
80
+ inputs = tokenizer.apply_chat_template(
81
+ messages,
82
+ tokenize = True,
83
+ add_generation_prompt = True,
84
+ return_tensors = "pt"
85
+ ).to("cuda")
86
+
87
+ # 3. Generate response
88
+ outputs = model.generate(input_ids = inputs, max_new_tokens = 512, use_cache = True)
89
+ print(tokenizer.decode(outputs[0]))
90
+ ```
91
 
92
  This qwen3_5 model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
93