Training in progress, epoch 1

adapter_config.json

@@ -19,13 +19,13 @@
   "rank_pattern": {},
   "revision": null,
   "target_modules": [
+    "v_proj",
+    "down_proj",
+    "q_proj",
     "up_proj",
-    "o_proj",
     "k_proj",
     "gate_proj",
-    "q_proj",
-    "down_proj",
-    "v_proj"
+    "o_proj"
   ],
   "task_type": "CAUSAL_LM",
   "use_rslora": false

adapter_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e92f82de5230f0325f712593d34e359445f49e7d9d62ef3fb29ecf1bbc991ad2
-size 1803907984
+oid sha256:7d5dfcb47d2492bcd381c74fff215f0c601db4a68fcf916e38070d7d97898d38
+size 1803940752

added_tokens.json

@@ -1,4 +1,6 @@
 {
-  "<|im_end|>": 32016,
-  "<|im_start|>": 32017
+  "DIALECT": 32018,
+  "[/DIALECT]": 32017,
+  "[PAD]": 32019,
+  "[SQL]": 32016
 }

special_tokens_map.json

@@ -1,28 +1,35 @@
 {
   "additional_special_tokens": [
     {
-      "content": "<|im_end|>",
+      "content": "[SQL]",
       "lstrip": false,
       "normalized": false,
       "rstrip": false,
       "single_word": false
     },
     {
-      "content": "<|im_start|>",
+      "content": "[/DIALECT]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false
+    },
+    {
+      "content": "DIALECT",
       "lstrip": false,
       "normalized": false,
       "rstrip": false,
       "single_word": false
     }
   ],
-  "bos_token": "<|im_start|>",
-  "eos_token": "<|im_end|>",
-  "pad_token": "<|im_end|>",
-  "unk_token": {
-    "content": "<unk>",
+  "bos_token": "<s>",
+  "eos_token": "</s>",
+  "pad_token": {
+    "content": "[PAD]",
     "lstrip": false,
     "normalized": false,
     "rstrip": false,
     "single_word": false
-  }
+  },
+  "unk_token": "<unk>"
 }

tmpdr2e1dme/_remote_module_non_scriptable.py

@@ -0,0 +1,81 @@
+from typing import *
+
+import torch
+import torch.distributed.rpc as rpc
+from torch import Tensor
+from torch._jit_internal import Future
+from torch.distributed.rpc import RRef
+from typing import Tuple  # pyre-ignore: unused import
+
+
+module_interface_cls = None
+
+
+def forward_async(self, *args, **kwargs):
+    args = (self.module_rref, self.device, self.is_device_map_set, *args)
+    kwargs = {**kwargs}
+    return rpc.rpc_async(
+        self.module_rref.owner(),
+        _remote_forward,
+        args,
+        kwargs,
+    )
+
+
+def forward(self, *args, **kwargs):
+    args = (self.module_rref, self.device, self.is_device_map_set, *args)
+    kwargs = {**kwargs}
+    ret_fut = rpc.rpc_async(
+        self.module_rref.owner(),
+        _remote_forward,
+        args,
+        kwargs,
+    )
+    return ret_fut.wait()
+
+
+_generated_methods = [
+    forward_async,
+    forward,
+]
+
+
+
+
+def _remote_forward(
+    module_rref: RRef[module_interface_cls], device: str, is_device_map_set: bool, *args, **kwargs):
+    module = module_rref.local_value()
+    device = torch.device(device)
+
+    if device.type != "cuda":
+        return module.forward(*args, **kwargs)
+
+    # If the module is on a cuda device,
+    # move any CPU tensor in args or kwargs to the same cuda device.
+    # Since torch script does not support generator expression,
+    # have to use concatenation instead of
+    # ``tuple(i.to(device) if isinstance(i, Tensor) else i for i in *args)``.
+    args = (*args,)
+    out_args: Tuple[()] = ()
+    for arg in args:
+        arg = (arg.to(device),) if isinstance(arg, Tensor) else (arg,)
+        out_args = out_args + arg
+
+    kwargs = {**kwargs}
+    for k, v in kwargs.items():
+        if isinstance(v, Tensor):
+            kwargs[k] = kwargs[k].to(device)
+
+    if is_device_map_set:
+        return module.forward(*out_args, **kwargs)
+
+    # If the device map is empty, then only CPU tensors are allowed to send over wire,
+    # so have to move any GPU tensor to CPU in the output.
+    # Since torch script does not support generator expression,
+    # have to use concatenation instead of
+    # ``tuple(i.cpu() if isinstance(i, Tensor) else i for i in module.forward(*out_args, **kwargs))``.
+    ret: Tuple[()] = ()
+    for i in module.forward(*out_args, **kwargs):
+        i = (i.cpu(),) if isinstance(i, Tensor) else (i,)
+        ret = ret + i
+    return ret

tokenizer.json

@@ -1,6 +1,11 @@
 {
   "version": "1.0",
-  "truncation": null,
+  "truncation": {
+    "direction": "Right",
+    "max_length": 3072,
+    "strategy": "LongestFirst",
+    "stride": 0
+  },
   "padding": null,
   "added_tokens": [
     {
@@ -68,7 +73,7 @@
     },
     {
       "id": 32016,
-      "content": "<|im_end|>",
+      "content": "[SQL]",
       "single_word": false,
       "lstrip": false,
       "rstrip": false,
@@ -77,7 +82,25 @@
     },
     {
       "id": 32017,
-      "content": "<|im_start|>",
+      "content": "[/DIALECT]",
+      "single_word": false,
+      "lstrip": false,
+      "rstrip": false,
+      "normalized": false,
+      "special": true
+    },
+    {
+      "id": 32018,
+      "content": "DIALECT",
+      "single_word": false,
+      "lstrip": false,
+      "rstrip": false,
+      "normalized": false,
+      "special": true
+    },
+    {
+      "id": 32019,
+      "content": "[PAD]",
       "single_word": false,
       "lstrip": false,
       "rstrip": false,

tokenizer_config.json

@@ -59,7 +59,7 @@
       "special": true
     },
     "32016": {
-      "content": "<|im_end|>",
+      "content": "[SQL]",
       "lstrip": false,
       "normalized": false,
       "rstrip": false,
@@ -67,7 +67,23 @@
       "special": true
     },
     "32017": {
-      "content": "<|im_start|>",
+      "content": "[/DIALECT]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32018": {
+      "content": "DIALECT",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32019": {
+      "content": "[PAD]",
       "lstrip": false,
       "normalized": false,
       "rstrip": false,
@@ -76,19 +92,19 @@
     }
   },
   "additional_special_tokens": [
-    "<|im_end|>",
-    "<|im_start|>"
+    "[SQL]",
+    "[/DIALECT]",
+    "DIALECT"
   ],
-  "bos_token": "<|im_start|>",
-  "chat_template": "{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
+  "bos_token": "<s>",
   "clean_up_tokenization_spaces": false,
-  "eos_token": "<|im_end|>",
+  "eos_token": "</s>",
   "eot_token": "▁<EOT>",
   "fill_token": "<FILL_ME>",
   "legacy": null,
   "middle_token": "▁<MID>",
   "model_max_length": 1000000000000000019884624838656,
-  "pad_token": "<|im_end|>",
+  "pad_token": "[PAD]",
   "prefix_token": "▁<PRE>",
   "sp_model_kwargs": {},
   "suffix_token": "▁<SUF>",

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4bafe043d2711fd6e68d65399d3d57e5bfff724c77431558500c810926dc46ea
+oid sha256:42e79e5a8a5acfb7f938b2a7d99d219ffaf4aa5de423917c0649be9238a74814
 size 4664