Upload folder using huggingface_hub

PosthogAgent/data/models/cluster_0_lora/README.md

@@ -0,0 +1,62 @@
+---
+base_model: mistralai/Mistral-7B-Instruct-v0.3
+library_name: peft
+model_name: cluster_0_lora
+tags:
+- base_model:adapter:mistralai/Mistral-7B-Instruct-v0.3
+- lora
+- sft
+- transformers
+- trl
+licence: license
+pipeline_tag: text-generation
+---
+
+# Model Card for cluster_0_lora
+
+This model is a fine-tuned version of [mistralai/Mistral-7B-Instruct-v0.3](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.3).
+It has been trained using [TRL](https://github.com/huggingface/trl).
+
+## Quick start
+
+```python
+from transformers import pipeline
+
+question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
+generator = pipeline("text-generation", model="None", device="cuda")
+output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
+print(output["generated_text"])
+```
+
+## Training procedure
+
+[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/amaan784-columbia-university/agentic-world/runs/octffxwk) 
+
+
+
+This model was trained with SFT.
+
+### Framework versions
+
+- PEFT 0.18.1
+- TRL: 0.29.0
+- Transformers: 5.2.0
+- Pytorch: 2.10.0
+- Datasets: 4.6.1
+- Tokenizers: 0.22.2
+
+## Citations
+
+
+
+Cite TRL as:
+    
+```bibtex
+@software{vonwerra2020trl,
+  title   = {{TRL: Transformers Reinforcement Learning}},
+  author  = {von Werra, Leandro and Belkada, Younes and Tunstall, Lewis and Beeching, Edward and Thrush, Tristan and Lambert, Nathan and Huang, Shengyi and Rasul, Kashif and Gallouédec, Quentin},
+  license = {Apache-2.0},
+  url     = {https://github.com/huggingface/trl},
+  year    = {2020}
+}
+```

PosthogAgent/data/models/cluster_0_lora/adapter_config.json

@@ -0,0 +1,46 @@
+{
+  "alora_invocation_tokens": null,
+  "alpha_pattern": {},
+  "arrow_config": null,
+  "auto_mapping": null,
+  "base_model_name_or_path": "mistralai/Mistral-7B-Instruct-v0.3",
+  "bias": "none",
+  "corda_config": null,
+  "ensure_weight_tying": false,
+  "eva_config": null,
+  "exclude_modules": null,
+  "fan_in_fan_out": false,
+  "inference_mode": true,
+  "init_lora_weights": true,
+  "layer_replication": null,
+  "layers_pattern": null,
+  "layers_to_transform": null,
+  "loftq_config": {},
+  "lora_alpha": 64,
+  "lora_bias": false,
+  "lora_dropout": 0.0,
+  "megatron_config": null,
+  "megatron_core": "megatron.core",
+  "modules_to_save": null,
+  "peft_type": "LORA",
+  "peft_version": "0.18.1",
+  "qalora_group_size": 16,
+  "r": 32,
+  "rank_pattern": {},
+  "revision": null,
+  "target_modules": [
+    "k_proj",
+    "down_proj",
+    "v_proj",
+    "gate_proj",
+    "up_proj",
+    "q_proj",
+    "o_proj"
+  ],
+  "target_parameters": null,
+  "task_type": "CAUSAL_LM",
+  "trainable_token_indices": null,
+  "use_dora": false,
+  "use_qalora": false,
+  "use_rslora": false
+}

PosthogAgent/data/models/cluster_0_lora/adapter_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8dd893f97bcdbd75c9d27f1be68b9223b946e763f3b07d1859e68fd9d306191b
+size 335604696

PosthogAgent/data/models/cluster_0_lora/chat_template.jinja

@@ -0,0 +1,87 @@
+{%- if messages[0]["role"] == "system" %}
+    {%- set system_message = messages[0]["content"] %}
+    {%- set loop_messages = messages[1:] %}
+{%- else %}
+    {%- set loop_messages = messages %}
+{%- endif %}
+{%- if not tools is defined %}
+    {%- set tools = none %}
+{%- endif %}
+{%- set user_messages = loop_messages | selectattr("role", "equalto", "user") | list %}
+
+{#- This block checks for alternating user/assistant messages, skipping tool calling messages #}
+{%- set ns = namespace() %}
+{%- set ns.index = 0 %}
+{%- for message in loop_messages %}
+    {%- if not (message.role == "tool" or message.role == "tool_results" or (message.tool_calls is defined and message.tool_calls is not none)) %}
+        {%- if (message["role"] == "user") != (ns.index % 2 == 0) %}
+            {{- raise_exception("After the optional system message, conversation roles must alternate user/assistant/user/assistant/...") }}
+        {%- endif %}
+        {%- set ns.index = ns.index + 1 %}
+    {%- endif %}
+{%- endfor %}
+
+{{- bos_token }}
+{%- for message in loop_messages %}
+    {%- if message["role"] == "user" %}
+        {%- if tools is not none and (message == user_messages[-1]) %}
+            {{- "[AVAILABLE_TOOLS] [" }}
+            {%- for tool in tools %}
+                {%- set tool = tool.function %}
+                {{- '{"type": "function", "function": {' }}
+                {%- for key, val in tool.items() if key != "return" %}
+                    {%- if val is string %}
+                        {{- '"' + key + '": "' + val + '"' }}
+                    {%- else %}
+                        {{- '"' + key + '": ' + val|tojson }}
+                    {%- endif %}
+                    {%- if not loop.last %}
+                        {{- ", " }}
+                    {%- endif %}
+                {%- endfor %}
+                {{- "}}" }}
+                {%- if not loop.last %}
+                    {{- ", " }}
+                {%- else %}
+                    {{- "]" }}
+                {%- endif %}
+            {%- endfor %}
+            {{- "[/AVAILABLE_TOOLS]" }}
+            {%- endif %}
+        {%- if loop.last and system_message is defined %}
+            {{- "[INST] " + system_message + "\n\n" + message["content"] + "[/INST]" }}
+        {%- else %}
+            {{- "[INST] " + message["content"] + "[/INST]" }}
+        {%- endif %}
+    {%- elif message.tool_calls is defined and message.tool_calls is not none %}
+        {{- "[TOOL_CALLS] [" }}
+        {%- for tool_call in message.tool_calls %}
+            {%- set out = tool_call.function|tojson %}
+            {{- out[:-1] }}
+            {%- if not tool_call.id is defined or tool_call.id|length != 9 %}
+                {{- raise_exception("Tool call IDs should be alphanumeric strings with length 9!") }}
+            {%- endif %}
+            {{- ', "id": "' + tool_call.id + '"}' }}
+            {%- if not loop.last %}
+                {{- ", " }}
+            {%- else %}
+                {{- "]" + eos_token }}
+            {%- endif %}
+        {%- endfor %}
+    {%- elif message["role"] == "assistant" %}
+        {{- " " + message["content"]|trim + eos_token}}
+    {%- elif message["role"] == "tool_results" or message["role"] == "tool" %}
+        {%- if message.content is defined and message.content.content is defined %}
+            {%- set content = message.content.content %}
+        {%- else %}
+            {%- set content = message.content %}
+        {%- endif %}
+        {{- '[TOOL_RESULTS] {"content": ' + content|string + ", " }}
+        {%- if not message.tool_call_id is defined or message.tool_call_id|length != 9 %}
+            {{- raise_exception("Tool call IDs should be alphanumeric strings with length 9!") }}
+        {%- endif %}
+        {{- '"call_id": "' + message.tool_call_id + '"}[/TOOL_RESULTS]' }}
+    {%- else %}
+        {{- raise_exception("Only user and assistant roles are supported, with the exception of an initial optional system message!") }}
+    {%- endif %}
+{%- endfor %}

PosthogAgent/data/models/cluster_0_lora/checkpoint-85/README.md

@@ -0,0 +1,209 @@
+---
+base_model: mistralai/Mistral-7B-Instruct-v0.3
+library_name: peft
+pipeline_tag: text-generation
+tags:
+- base_model:adapter:mistralai/Mistral-7B-Instruct-v0.3
+- lora
+- sft
+- transformers
+- trl
+---
+
+# Model Card for Model ID
+
+<!-- Provide a quick summary of what the model is/does. -->
+
+
+
+## Model Details
+
+### Model Description
+
+<!-- Provide a longer summary of what this model is. -->
+
+
+
+- **Developed by:** [More Information Needed]
+- **Funded by [optional]:** [More Information Needed]
+- **Shared by [optional]:** [More Information Needed]
+- **Model type:** [More Information Needed]
+- **Language(s) (NLP):** [More Information Needed]
+- **License:** [More Information Needed]
+- **Finetuned from model [optional]:** [More Information Needed]
+
+### Model Sources [optional]
+
+<!-- Provide the basic links for the model. -->
+
+- **Repository:** [More Information Needed]
+- **Paper [optional]:** [More Information Needed]
+- **Demo [optional]:** [More Information Needed]
+
+## Uses
+
+<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
+
+### Direct Use
+
+<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
+
+[More Information Needed]
+
+### Downstream Use [optional]
+
+<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
+
+[More Information Needed]
+
+### Out-of-Scope Use
+
+<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
+
+[More Information Needed]
+
+## Bias, Risks, and Limitations
+
+<!-- This section is meant to convey both technical and sociotechnical limitations. -->
+
+[More Information Needed]
+
+### Recommendations
+
+<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
+
+Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
+
+## How to Get Started with the Model
+
+Use the code below to get started with the model.
+
+[More Information Needed]
+
+## Training Details
+
+### Training Data
+
+<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
+
+[More Information Needed]
+
+### Training Procedure
+
+<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
+
+#### Preprocessing [optional]
+
+[More Information Needed]
+
+
+#### Training Hyperparameters
+
+- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
+
+#### Speeds, Sizes, Times [optional]
+
+<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
+
+[More Information Needed]
+
+## Evaluation
+
+<!-- This section describes the evaluation protocols and provides the results. -->
+
+### Testing Data, Factors & Metrics
+
+#### Testing Data
+
+<!-- This should link to a Dataset Card if possible. -->
+
+[More Information Needed]
+
+#### Factors
+
+<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
+
+[More Information Needed]
+
+#### Metrics
+
+<!-- These are the evaluation metrics being used, ideally with a description of why. -->
+
+[More Information Needed]
+
+### Results
+
+[More Information Needed]
+
+#### Summary
+
+
+
+## Model Examination [optional]
+
+<!-- Relevant interpretability work for the model goes here -->
+
+[More Information Needed]
+
+## Environmental Impact
+
+<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
+
+Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
+
+- **Hardware Type:** [More Information Needed]
+- **Hours used:** [More Information Needed]
+- **Cloud Provider:** [More Information Needed]
+- **Compute Region:** [More Information Needed]
+- **Carbon Emitted:** [More Information Needed]
+
+## Technical Specifications [optional]
+
+### Model Architecture and Objective
+
+[More Information Needed]
+
+### Compute Infrastructure
+
+[More Information Needed]
+
+#### Hardware
+
+[More Information Needed]
+
+#### Software
+
+[More Information Needed]
+
+## Citation [optional]
+
+<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
+
+**BibTeX:**
+
+[More Information Needed]
+
+**APA:**
+
+[More Information Needed]
+
+## Glossary [optional]
+
+<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
+
+[More Information Needed]
+
+## More Information [optional]
+
+[More Information Needed]
+
+## Model Card Authors [optional]
+
+[More Information Needed]
+
+## Model Card Contact
+
+[More Information Needed]
+### Framework versions
+
+- PEFT 0.18.1

PosthogAgent/data/models/cluster_0_lora/checkpoint-85/adapter_config.json

@@ -0,0 +1,46 @@
+{
+  "alora_invocation_tokens": null,
+  "alpha_pattern": {},
+  "arrow_config": null,
+  "auto_mapping": null,
+  "base_model_name_or_path": "mistralai/Mistral-7B-Instruct-v0.3",
+  "bias": "none",
+  "corda_config": null,
+  "ensure_weight_tying": false,
+  "eva_config": null,
+  "exclude_modules": null,
+  "fan_in_fan_out": false,
+  "inference_mode": true,
+  "init_lora_weights": true,
+  "layer_replication": null,
+  "layers_pattern": null,
+  "layers_to_transform": null,
+  "loftq_config": {},
+  "lora_alpha": 64,
+  "lora_bias": false,
+  "lora_dropout": 0.0,
+  "megatron_config": null,
+  "megatron_core": "megatron.core",
+  "modules_to_save": null,
+  "peft_type": "LORA",
+  "peft_version": "0.18.1",
+  "qalora_group_size": 16,
+  "r": 32,
+  "rank_pattern": {},
+  "revision": null,
+  "target_modules": [
+    "k_proj",
+    "down_proj",
+    "v_proj",
+    "gate_proj",
+    "up_proj",
+    "q_proj",
+    "o_proj"
+  ],
+  "target_parameters": null,
+  "task_type": "CAUSAL_LM",
+  "trainable_token_indices": null,
+  "use_dora": false,
+  "use_qalora": false,
+  "use_rslora": false
+}

PosthogAgent/data/models/cluster_0_lora/checkpoint-85/adapter_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8dd893f97bcdbd75c9d27f1be68b9223b946e763f3b07d1859e68fd9d306191b
+size 335604696

PosthogAgent/data/models/cluster_0_lora/checkpoint-85/chat_template.jinja

@@ -0,0 +1,87 @@
+{%- if messages[0]["role"] == "system" %}
+    {%- set system_message = messages[0]["content"] %}
+    {%- set loop_messages = messages[1:] %}
+{%- else %}
+    {%- set loop_messages = messages %}
+{%- endif %}
+{%- if not tools is defined %}
+    {%- set tools = none %}
+{%- endif %}
+{%- set user_messages = loop_messages | selectattr("role", "equalto", "user") | list %}
+
+{#- This block checks for alternating user/assistant messages, skipping tool calling messages #}
+{%- set ns = namespace() %}
+{%- set ns.index = 0 %}
+{%- for message in loop_messages %}
+    {%- if not (message.role == "tool" or message.role == "tool_results" or (message.tool_calls is defined and message.tool_calls is not none)) %}
+        {%- if (message["role"] == "user") != (ns.index % 2 == 0) %}
+            {{- raise_exception("After the optional system message, conversation roles must alternate user/assistant/user/assistant/...") }}
+        {%- endif %}
+        {%- set ns.index = ns.index + 1 %}
+    {%- endif %}
+{%- endfor %}
+
+{{- bos_token }}
+{%- for message in loop_messages %}
+    {%- if message["role"] == "user" %}
+        {%- if tools is not none and (message == user_messages[-1]) %}
+            {{- "[AVAILABLE_TOOLS] [" }}
+            {%- for tool in tools %}
+                {%- set tool = tool.function %}
+                {{- '{"type": "function", "function": {' }}
+                {%- for key, val in tool.items() if key != "return" %}
+                    {%- if val is string %}
+                        {{- '"' + key + '": "' + val + '"' }}
+                    {%- else %}
+                        {{- '"' + key + '": ' + val|tojson }}
+                    {%- endif %}
+                    {%- if not loop.last %}
+                        {{- ", " }}
+                    {%- endif %}
+                {%- endfor %}
+                {{- "}}" }}
+                {%- if not loop.last %}
+                    {{- ", " }}
+                {%- else %}
+                    {{- "]" }}
+                {%- endif %}
+            {%- endfor %}
+            {{- "[/AVAILABLE_TOOLS]" }}
+            {%- endif %}
+        {%- if loop.last and system_message is defined %}
+            {{- "[INST] " + system_message + "\n\n" + message["content"] + "[/INST]" }}
+        {%- else %}
+            {{- "[INST] " + message["content"] + "[/INST]" }}
+        {%- endif %}
+    {%- elif message.tool_calls is defined and message.tool_calls is not none %}
+        {{- "[TOOL_CALLS] [" }}
+        {%- for tool_call in message.tool_calls %}
+            {%- set out = tool_call.function|tojson %}
+            {{- out[:-1] }}
+            {%- if not tool_call.id is defined or tool_call.id|length != 9 %}
+                {{- raise_exception("Tool call IDs should be alphanumeric strings with length 9!") }}
+            {%- endif %}
+            {{- ', "id": "' + tool_call.id + '"}' }}
+            {%- if not loop.last %}
+                {{- ", " }}
+            {%- else %}
+                {{- "]" + eos_token }}
+            {%- endif %}
+        {%- endfor %}
+    {%- elif message["role"] == "assistant" %}
+        {{- " " + message["content"]|trim + eos_token}}
+    {%- elif message["role"] == "tool_results" or message["role"] == "tool" %}
+        {%- if message.content is defined and message.content.content is defined %}
+            {%- set content = message.content.content %}
+        {%- else %}
+            {%- set content = message.content %}
+        {%- endif %}
+        {{- '[TOOL_RESULTS] {"content": ' + content|string + ", " }}
+        {%- if not message.tool_call_id is defined or message.tool_call_id|length != 9 %}
+            {{- raise_exception("Tool call IDs should be alphanumeric strings with length 9!") }}
+        {%- endif %}
+        {{- '"call_id": "' + message.tool_call_id + '"}[/TOOL_RESULTS]' }}
+    {%- else %}
+        {{- raise_exception("Only user and assistant roles are supported, with the exception of an initial optional system message!") }}
+    {%- endif %}
+{%- endfor %}

PosthogAgent/data/models/cluster_0_lora/checkpoint-85/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d781c9773911b562e2984b08693b71873e9eba24567a2d53a032b4ea4ed89216
+size 671473443

PosthogAgent/data/models/cluster_0_lora/checkpoint-85/rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:17cd930da9783ca70bad4b9cdeee6a06c0acea8f34645a333c93341f487f66a3
+size 14645

PosthogAgent/data/models/cluster_0_lora/checkpoint-85/scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c6a6ae75236fc6775e769748d1d9884d2915a6f97c9abf3085c78547fb15d123
+size 1465

PosthogAgent/data/models/cluster_0_lora/checkpoint-85/tokenizer_config.json

@@ -0,0 +1,16 @@
+{
+  "add_prefix_space": true,
+  "backend": "tokenizers",
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "</s>",
+  "is_local": false,
+  "legacy": false,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "</s>",
+  "sp_model_kwargs": {},
+  "spaces_between_special_tokens": false,
+  "tokenizer_class": "TokenizersBackend",
+  "unk_token": "<unk>",
+  "use_default_system_prompt": false
+}

PosthogAgent/data/models/cluster_0_lora/checkpoint-85/trainer_state.json

@@ -0,0 +1,259 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 5.0,
+  "eval_steps": 500,
+  "global_step": 85,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "entropy": 1.8334363281726838,
+      "epoch": 0.29411764705882354,
+      "grad_norm": 1.8545987606048584,
+      "learning_rate": 0.000199926618105081,
+      "loss": 2.2091386795043944,
+      "mean_token_accuracy": 0.649028092622757,
+      "num_tokens": 155901.0,
+      "step": 5
+    },
+    {
+      "entropy": 0.5176878422498703,
+      "epoch": 0.5882352941176471,
+      "grad_norm": 0.3128144443035126,
+      "learning_rate": 0.00019736954238777792,
+      "loss": 0.6189306259155274,
+      "mean_token_accuracy": 0.8548313289880752,
+      "num_tokens": 314767.0,
+      "step": 10
+    },
+    {
+      "entropy": 0.3833584442734718,
+      "epoch": 0.8823529411764706,
+      "grad_norm": 0.22035302221775055,
+      "learning_rate": 0.000191250361647655,
+      "loss": 0.4222111701965332,
+      "mean_token_accuracy": 0.8876540273427963,
+      "num_tokens": 471502.0,
+      "step": 15
+    },
+    {
+      "epoch": 1.0,
+      "eval_entropy": 0.3990151807665825,
+      "eval_loss": 0.35258573293685913,
+      "eval_mean_token_accuracy": 0.8936482965946198,
+      "eval_num_tokens": 531236.0,
+      "eval_runtime": 6.1597,
+      "eval_samples_per_second": 5.195,
+      "eval_steps_per_second": 0.649,
+      "step": 17
+    },
+    {
+      "entropy": 0.39187141358852384,
+      "epoch": 1.1764705882352942,
+      "grad_norm": 0.13372336328029633,
+      "learning_rate": 0.00018179293607667178,
+      "loss": 0.3483515739440918,
+      "mean_token_accuracy": 0.8957118839025497,
+      "num_tokens": 626243.0,
+      "step": 20
+    },
+    {
+      "entropy": 0.3649876207113266,
+      "epoch": 1.4705882352941178,
+      "grad_norm": 0.13568232953548431,
+      "learning_rate": 0.00016934325007922417,
+      "loss": 0.3126405715942383,
+      "mean_token_accuracy": 0.9013186633586884,
+      "num_tokens": 783681.0,
+      "step": 25
+    },
+    {
+      "entropy": 0.3379180431365967,
+      "epoch": 1.7647058823529411,
+      "grad_norm": 0.08893899619579315,
+      "learning_rate": 0.00015435675500012212,
+      "loss": 0.2885303020477295,
+      "mean_token_accuracy": 0.9048597484827041,
+      "num_tokens": 939886.0,
+      "step": 30
+    },
+    {
+      "epoch": 2.0,
+      "eval_entropy": 0.3233979865908623,
+      "eval_loss": 0.285524845123291,
+      "eval_mean_token_accuracy": 0.9056155234575272,
+      "eval_num_tokens": 1062472.0,
+      "eval_runtime": 6.1561,
+      "eval_samples_per_second": 5.198,
+      "eval_steps_per_second": 0.65,
+      "step": 34
+    },
+    {
+      "entropy": 0.32396653294563293,
+      "epoch": 2.0588235294117645,
+      "grad_norm": 0.09661662578582764,
+      "learning_rate": 0.00013738170718407687,
+      "loss": 0.2859357833862305,
+      "mean_token_accuracy": 0.9059436947107316,
+      "num_tokens": 1094321.0,
+      "step": 35
+    },
+    {
+      "entropy": 0.3076239213347435,
+      "epoch": 2.3529411764705883,
+      "grad_norm": 0.07090552151203156,
+      "learning_rate": 0.00011903911091646684,
+      "loss": 0.2774367094039917,
+      "mean_token_accuracy": 0.9083002626895904,
+      "num_tokens": 1252927.0,
+      "step": 40
+    },
+    {
+      "entropy": 0.31161843091249464,
+      "epoch": 2.6470588235294117,
+      "grad_norm": 0.06575407832860947,
+      "learning_rate": 0.0001,
+      "loss": 0.2757134914398193,
+      "mean_token_accuracy": 0.909886708855629,
+      "num_tokens": 1411065.0,
+      "step": 45
+    },
+    {
+      "entropy": 0.2939224377274513,
+      "epoch": 2.9411764705882355,
+      "grad_norm": 0.07291047275066376,
+      "learning_rate": 8.096088908353315e-05,
+      "loss": 0.2671217679977417,
+      "mean_token_accuracy": 0.9137494266033173,
+      "num_tokens": 1566441.0,
+      "step": 50
+    },
+    {
+      "epoch": 3.0,
+      "eval_entropy": 0.3115631192922592,
+      "eval_loss": 0.27641046047210693,
+      "eval_mean_token_accuracy": 0.9095478951931,
+      "eval_num_tokens": 1593708.0,
+      "eval_runtime": 6.1474,
+      "eval_samples_per_second": 5.205,
+      "eval_steps_per_second": 0.651,
+      "step": 51
+    },
+    {
+      "entropy": 0.30940164178609847,
+      "epoch": 3.235294117647059,
+      "grad_norm": 0.06909104436635971,
+      "learning_rate": 6.261829281592314e-05,
+      "loss": 0.2742582082748413,
+      "mean_token_accuracy": 0.9103405773639679,
+      "num_tokens": 1720754.0,
+      "step": 55
+    },
+    {
+      "entropy": 0.29637852013111116,
+      "epoch": 3.5294117647058822,
+      "grad_norm": 0.05777778849005699,
+      "learning_rate": 4.56432449998779e-05,
+      "loss": 0.2619441747665405,
+      "mean_token_accuracy": 0.9152446120977402,
+      "num_tokens": 1877258.0,
+      "step": 60
+    },
+    {
+      "entropy": 0.29939648061990737,
+      "epoch": 3.8235294117647056,
+      "grad_norm": 0.0680684745311737,
+      "learning_rate": 3.065674992077584e-05,
+      "loss": 0.26411483287811277,
+      "mean_token_accuracy": 0.9143152892589569,
+      "num_tokens": 2033336.0,
+      "step": 65
+    },
+    {
+      "epoch": 4.0,
+      "eval_entropy": 0.2924557253718376,
+      "eval_loss": 0.2721189260482788,
+      "eval_mean_token_accuracy": 0.911041259765625,
+      "eval_num_tokens": 2124944.0,
+      "eval_runtime": 6.1524,
+      "eval_samples_per_second": 5.201,
+      "eval_steps_per_second": 0.65,
+      "step": 68
+    },
+    {
+      "entropy": 0.29475061744451525,
+      "epoch": 4.117647058823529,
+      "grad_norm": 0.05317465960979462,
+      "learning_rate": 1.8207063923328237e-05,
+      "loss": 0.2643876075744629,
+      "mean_token_accuracy": 0.9139553159475327,
+      "num_tokens": 2188076.0,
+      "step": 70
+    },
+    {
+      "entropy": 0.2941878780722618,
+      "epoch": 4.411764705882353,
+      "grad_norm": 0.0484347939491272,
+      "learning_rate": 8.749638352345002e-06,
+      "loss": 0.2619620323181152,
+      "mean_token_accuracy": 0.9148574978113174,
+      "num_tokens": 2345481.0,
+      "step": 75
+    },
+    {
+      "entropy": 0.29366316497325895,
+      "epoch": 4.705882352941177,
+      "grad_norm": 0.05014990270137787,
+      "learning_rate": 2.6304576122221035e-06,
+      "loss": 0.2590233087539673,
+      "mean_token_accuracy": 0.9161737859249115,
+      "num_tokens": 2501922.0,
+      "step": 80
+    },
+    {
+      "entropy": 0.29574155658483503,
+      "epoch": 5.0,
+      "grad_norm": 0.05177079513669014,
+      "learning_rate": 7.338189491900016e-08,
+      "loss": 0.2599493980407715,
+      "mean_token_accuracy": 0.916091114282608,
+      "num_tokens": 2656180.0,
+      "step": 85
+    },
+    {
+      "epoch": 5.0,
+      "eval_entropy": 0.29599621891975403,
+      "eval_loss": 0.27076640725135803,
+      "eval_mean_token_accuracy": 0.9115611612796783,
+      "eval_num_tokens": 2656180.0,
+      "eval_runtime": 6.1568,
+      "eval_samples_per_second": 5.198,
+      "eval_steps_per_second": 0.65,
+      "step": 85
+    }
+  ],
+  "logging_steps": 5,
+  "max_steps": 85,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 5,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 1.1471019168940032e+17,
+  "train_batch_size": 4,
+  "trial_name": null,
+  "trial_params": null
+}

PosthogAgent/data/models/cluster_0_lora/checkpoint-85/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:64bb22527a4f5c27eb109a9344cccfbcca0a6c72193206a445ba784bef97b1d1
+size 5713

PosthogAgent/data/models/cluster_0_lora/tokenizer_config.json

@@ -0,0 +1,16 @@
+{
+  "add_prefix_space": true,
+  "backend": "tokenizers",
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "</s>",
+  "is_local": false,
+  "legacy": false,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "</s>",
+  "sp_model_kwargs": {},
+  "spaces_between_special_tokens": false,
+  "tokenizer_class": "TokenizersBackend",
+  "unk_token": "<unk>",
+  "use_default_system_prompt": false
+}

PosthogAgent/data/models/cluster_0_lora/training_config.json

@@ -0,0 +1,16 @@
+{
+  "cluster_id": 0,
+  "cluster_label": "Focused Speedster",
+  "base_model": "mistralai/Mistral-7B-Instruct-v0.3",
+  "lora_rank": 32,
+  "lora_alpha": 64,
+  "epochs": 5,
+  "learning_rate": 0.0002,
+  "max_seq_length": 2048,
+  "train_examples": 2486,
+  "eval_examples": 276,
+  "final_train_loss": 0.42068530811982996,
+  "final_eval_loss": 0.27076640725135803,
+  "training_time_s": 826.985,
+  "peak_vram_gb": 21.318359375
+}