marisming commited on
Commit
a586ace
·
verified ·
1 Parent(s): f468fc9

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +128 -0
README.md ADDED
@@ -0,0 +1,128 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ tags:
6
+ - biology
7
+ - protein
8
+ - bioinformatics
9
+ - mixture-of-experts
10
+ - gguf
11
+ - llama.cpp
12
+ base_model: dnagpt/OmniGene-4-CPT-v2-merged
13
+ quantized_by: Liang Wang
14
+ ---
15
+
16
+ # OmniGene-4-CPT-v2-GGUF
17
+
18
+ **GGUF format models for OmniGene-4-CPT-v2** (continued pretraining checkpoint)
19
+
20
+ GGUF format quantized versions of OmniGene-4 for efficient inference on consumer GPUs and CPUs using llama.cpp, llama-cpp-python, Ollama, LM Studio, and other GGUF-compatible runtimes.
21
+
22
+ ## Available Quantizations
23
+
24
+ | Quantization | File | Size | RAM Required | Quality |
25
+ |---|---|---|---|---|
26
+ | **F16** | `OmniGene-4-CPT-v2-f16.gguf` | 50.6 GB | ~52 GB | Best quality |
27
+ | **Q4_K_M** | `OmniGene-4-CPT-v2-Q4_K_M.gguf` | 16 GB | ~17 GB | Recommended balance |
28
+
29
+ ## Hardware Requirements
30
+
31
+ | Quantization | GPU | CPU + RAM |
32
+ |---|---|---|
33
+ | **F16** | RTX A6000 (48GB) | 64GB+ system RAM |
34
+ | **Q4_K_M** | RTX 5090 (32GB) / RTX 4090 (24GB) / RTX 3090 (24GB) | 32GB+ system RAM |
35
+
36
+ ## Quick Start
37
+
38
+ ### Option 1: llama-cpp-python
39
+
40
+ ```bash
41
+ pip install llama-cpp-python
42
+ ```
43
+
44
+ ```python
45
+ from llama_cpp import Llama
46
+
47
+ llm = Llama(
48
+ model_path="OmniGene-4-CPT-v2-Q4_K_M.gguf",
49
+ n_ctx=4096,
50
+ n_gpu_layers=-1, # Offload all layers to GPU
51
+ )
52
+
53
+ output = llm("MKTAYIAKQRQISFVKSHFSRQLEERLGLIEVQAPILSRVGDGTQDNLSGAEK", max_tokens=100)
54
+ print(output['choices'][0]['text'])
55
+ ```
56
+
57
+ ### Option 2: llama.cpp Command Line
58
+
59
+ ```bash
60
+ ./llama-cli -m OmniGene-4-CPT-v2-Q4_K_M.gguf -p "MKTAYIAKQRQISFVKSHFSRQLEERL" -n 100 -ngl -1
61
+ ```
62
+
63
+ ### Option 3: Ollama
64
+
65
+ ```bash
66
+ # Create Modelfile
67
+ cat > Modelfile <<EOF
68
+ FROM ./OmniGene-4-CPT-v2-Q4_K_M.gguf
69
+ EOF
70
+
71
+ ollama create omnigene-4-cpt -f Modelfile
72
+ ollama run omnigene-4-cpt
73
+ ```
74
+
75
+ ### Option 4: LM Studio
76
+
77
+ 1. Download `OmniGene-4-CPT-v2-Q4_K_M.gguf`
78
+ 2. Place in LM Studio models folder
79
+ 3. Load in LM Studio
80
+ 4. Start chatting
81
+
82
+ ## Model Description
83
+
84
+ OmniGene-4-CPT-v2 is a biological foundation model with:
85
+ - **Base**: Gemma-4-26B-A4B-Instruct (MoE, 128 experts, top-8 routing)
86
+ - **Vocabulary**: 290,048 tokens (262,020 original + 28,028 bio tokens)
87
+ - **CPT data**: 32.5 GB mixed corpus (DNA, Protein, OpenWebText, Structure)
88
+ - **Training**: 0.6 epoch, 2,806 steps, 8×H20 GPUs
89
+
90
+ ## Biological Tokens
91
+
92
+ The model includes 28,028 additional biological tokens:
93
+ - **DNA BPE**: 20,000 tokens (optimized for genomic sequences)
94
+ - **Protein BPE**: 8,000 tokens (optimized for amino acid sequences)
95
+ - **3Di alphabet**: 20 tokens (Foldseek structural alphabet)
96
+ - **DSSP**: 8 tokens (secondary structure: H, E, C, etc.)
97
+
98
+ ## Other Versions
99
+
100
+ - **Full BF16** (HuggingFace transformers): https://huggingface.co/dnagpt/OmniGene-4-CPT-v2-merged
101
+ - **LoRA adapter** (requires base model): https://huggingface.co/dnagpt/OmniGene-4-CPT-v2
102
+ - **4-bit auto-quantize**: https://huggingface.co/dnagpt/OmniGene-4-CPT-v2-4bit
103
+ - **Instruction-tuned GGUF**: https://huggingface.co/dnagpt/OmniGene-4-SFT-v3-GGUF
104
+
105
+ ## Citation
106
+
107
+ ```bibtex
108
+ @article{wang2026omnigene4,
109
+ title={OmniGene-4: A Unified Bio-Language MoE Model with Router-Level Interpretability},
110
+ author={Wang, Liang},
111
+ journal={bioRxiv},
112
+ year={2026}
113
+ }
114
+ ```
115
+
116
+ ## Paper
117
+
118
+ Full paper: https://github.com/maris205/omnigene4
119
+
120
+ ## License
121
+
122
+ Apache 2.0
123
+
124
+ ## Contact
125
+
126
+ Liang Wang (wangliang.f@gmail.com)
127
+ School of Artificial Intelligence and Automation
128
+ Huazhong University of Science and Technology