King3Djbl commited on
Commit
7b31645
·
verified ·
1 Parent(s): 5eb70c5

Add model card and config files for FableForge

Browse files
README.md ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ license: mit
5
+ library_name: transformers
6
+ pipeline_tag: text-generation
7
+ tags:
8
+ - fableforge
9
+ - agent
10
+ - code-generation
11
+ - tool-use
12
+ - reasoning
13
+ - base
14
+ base_model: meta-llama/Llama-2-7b-chat-hf
15
+ ---
16
+
17
+ # FableForge
18
+
19
+ The base unified agent model - a 7B parameter model fine-tuned for agent tasks including planning, tool use, code generation, and self-correction. The foundation model for the FableForge ecosystem.
20
+
21
+ ## Quick Start
22
+
23
+ ```python
24
+ from transformers import AutoModelForCausalLM, AutoTokenizer
25
+
26
+ model_name = "fableforge-ai/FableForge"
27
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
28
+ model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
29
+
30
+ prompt = """You are an AI agent. Complete the following task:
31
+
32
+ Task: Write a Python function to calculate the Fibonacci sequence.
33
+
34
+ Reasoning:"""
35
+
36
+ inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
37
+ outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.6, top_p=0.9)
38
+ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
39
+ ```
40
+
41
+ ## Use Cases
42
+
43
+ - General-purpose agent tasks
44
+ - Planning and decomposition
45
+ - Code generation with self-verification
46
+ - Integration with FableForge runtime and tools
47
+
48
+ ### Integration with FableForge Ecosystem
49
+
50
+ ```python
51
+ from fableforge_agent_runtime import AgentRuntime
52
+ from fableforge_agent_skills import SkillLibrary
53
+
54
+ runtime = AgentRuntime(
55
+ model="fableforge-ai/FableForge",
56
+ skills=SkillLibrary.all(),
57
+ verification=True
58
+ )
59
+
60
+ result = runtime.run("Deploy a web server on AWS")
61
+ print(result.output)
62
+ print(result.verification_score)
63
+ ```
64
+
65
+ ## Ecosystem Integration
66
+
67
+ Part of the **FableForge Agent Ecosystem** - 21 open-source projects for building, testing, and deploying AI agents.
68
+
69
+ | Package | Install | Purpose |
70
+ |---------|---------|---------|
71
+ | `fableforge` | `pip install fableforge` | Unified CLI |
72
+ | `fableforge-anvil-agent` | `pip install fableforge-anvil-agent` | Self-verified coding agent |
73
+ | `fableforge-agent-swarm` | `pip install fableforge-agent-swarm` | Multi-agent orchestration |
74
+ | `fableforge-agent-runtime` | `pip install fableforge-agent-runtime` | Production agent runtime |
75
+ | `fableforge-agent-skills` | `pip install fableforge-agent-skills` | Skill library |
76
+ | `verifyloop` | `pip install verifyloop` | Verification loops |
77
+ | `reason-critic` | `pip install reason-critic` | Reasoning assessment |
78
+
79
+ ## Model Details
80
+
81
+ | Attribute | Value |
82
+ |-----------|-------|
83
+ | Architecture | LlamaForCausalLM |
84
+ | Parameters | 7B |
85
+ | Hidden Size | 4096 |
86
+ | Layers | 32 |
87
+ | Attention Heads | 32 |
88
+ | KV Heads | 32 |
89
+ | Max Context | 4096 |
90
+ | Training Data | Fable5 agent traces + curated reasoning datasets |
91
+ | License | MIT |
92
+
93
+ ## Limitations
94
+
95
+ - May generate incorrect code -- always use with verifyloop for critical tasks
96
+ - Trained primarily on English data; multilingual performance is limited
97
+ - Can hallucinate API signatures or tool parameters
98
+ - Not suitable for medical, legal, or financial advice without human review
99
+
100
+ ## Citation
101
+
102
+ ```bibtex
103
+ @misc{fableforge2024,
104
+ title={FableForge: Agent Orchestration via Fine-Tuned Language Models},
105
+ author={FableForge Team},
106
+ year={2024},
107
+ url={https://huggingface.co/fableforge-ai/FableForge}
108
+ }
109
+ ```
110
+
111
+ ## License
112
+
113
+ MIT License - see [LICENSE](LICENSE) for details.
114
+
115
+ ---
116
+
117
+ Built with hammer by the [FableForge](https://github.com/KingLabsA) team. Part of the [FableForge ecosystem](https://kinglabsa.github.io/fableforge/).
config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "LlamaForCausalLM"
4
+ ],
5
+ "model_type": "llama",
6
+ "hidden_size": 4096,
7
+ "intermediate_size": 11008,
8
+ "num_hidden_layers": 32,
9
+ "num_attention_heads": 32,
10
+ "num_key_value_heads": 32,
11
+ "vocab_size": 32000,
12
+ "max_position_embeddings": 4096,
13
+ "rms_norm_eps": 1e-05,
14
+ "rope_theta": 10000.0,
15
+ "tie_word_embeddings": false,
16
+ "torch_dtype": "float16",
17
+ "use_cache": true,
18
+ "bos_token_id": 1,
19
+ "eos_token_id": 2,
20
+ "pad_token_id": 0
21
+ }
generation_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 1,
3
+ "eos_token_id": 2,
4
+ "do_sample": true,
5
+ "temperature": 0.6,
6
+ "top_p": 0.9,
7
+ "top_k": 50,
8
+ "repetition_penalty": 1.1,
9
+ "max_new_tokens": 2048
10
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<s>",
3
+ "eos_token": "</s>",
4
+ "unk_token": "<unk>",
5
+ "pad_token": "<pad>",
6
+ "sep_token": "</s>",
7
+ "cls_token": "<s>",
8
+ "mask_token": "<mask>"
9
+ }
tokenizer.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "version": "1.0.0",
3
+ "truncation": null,
4
+ "padding": null,
5
+ "added_tokens": [
6
+ {
7
+ "id": 0,
8
+ "content": "<unk>",
9
+ "single_word": false,
10
+ "lstrip": false,
11
+ "rstrip": false,
12
+ "normalized": false,
13
+ "special": true
14
+ },
15
+ {
16
+ "id": 1,
17
+ "content": "<s>",
18
+ "single_word": false,
19
+ "lstrip": false,
20
+ "rstrip": false,
21
+ "normalized": false,
22
+ "special": true
23
+ },
24
+ {
25
+ "id": 2,
26
+ "content": "</s>",
27
+ "single_word": false,
28
+ "lstrip": false,
29
+ "rstrip": false,
30
+ "normalized": false,
31
+ "special": true
32
+ }
33
+ ],
34
+ "normalizer": null,
35
+ "pre_tokenizer": {
36
+ "type": "ByteLevel",
37
+ "add_prefix_space": false,
38
+ "trim_offsets": true,
39
+ "use_regex": true
40
+ },
41
+ "post_processor": {
42
+ "type": "ByteLevel",
43
+ "add_prefix_space": true,
44
+ "trim_offsets": false,
45
+ "use_regex": true
46
+ },
47
+ "decoder": {
48
+ "type": "ByteLevel"
49
+ },
50
+ "model": {
51
+ "type": "BPE",
52
+ "dropout": null,
53
+ "unk_token": "<unk>",
54
+ "continuing_subword_prefix": null,
55
+ "end_of_word_suffix": null,
56
+ "fuse_unk": false,
57
+ "byte_fallback": false,
58
+ "vocab": {
59
+ "<unk>": 0,
60
+ "<s>": 1,
61
+ "</s>": 2
62
+ },
63
+ "merges": []
64
+ }
65
+ }
tokenizer_config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": false,
5
+ "bos_token": {
6
+ "content": "<s>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false
11
+ },
12
+ "eos_token": {
13
+ "content": "</s>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false
18
+ },
19
+ "unk_token": {
20
+ "content": "<unk>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false
25
+ },
26
+ "pad_token": {
27
+ "content": "<pad>",
28
+ "lstrip": false,
29
+ "normalized": false,
30
+ "rstrip": false,
31
+ "single_word": false
32
+ },
33
+ "model_type": "llama",
34
+ "model_max_length": 4096,
35
+ "tokenizer_class": "PreTrainedTokenizerFast",
36
+ "clean_up_tokenization_spaces": false
37
+ }