Upload 9 files

added_tokens.json

@@ -0,0 +1,145 @@
+{
+  "<|extratoken_100|>": 50356,
+  "<|extratoken_101|>": 50357,
+  "<|extratoken_102|>": 50358,
+  "<|extratoken_103|>": 50359,
+  "<|extratoken_104|>": 50360,
+  "<|extratoken_105|>": 50361,
+  "<|extratoken_106|>": 50362,
+  "<|extratoken_107|>": 50363,
+  "<|extratoken_108|>": 50364,
+  "<|extratoken_109|>": 50365,
+  "<|extratoken_10|>": 50266,
+  "<|extratoken_110|>": 50366,
+  "<|extratoken_111|>": 50367,
+  "<|extratoken_112|>": 50368,
+  "<|extratoken_113|>": 50369,
+  "<|extratoken_114|>": 50370,
+  "<|extratoken_115|>": 50371,
+  "<|extratoken_116|>": 50372,
+  "<|extratoken_117|>": 50373,
+  "<|extratoken_118|>": 50374,
+  "<|extratoken_119|>": 50375,
+  "<|extratoken_11|>": 50267,
+  "<|extratoken_120|>": 50376,
+  "<|extratoken_121|>": 50377,
+  "<|extratoken_122|>": 50378,
+  "<|extratoken_123|>": 50379,
+  "<|extratoken_124|>": 50380,
+  "<|extratoken_125|>": 50381,
+  "<|extratoken_126|>": 50382,
+  "<|extratoken_127|>": 50383,
+  "<|extratoken_128|>": 50384,
+  "<|extratoken_129|>": 50385,
+  "<|extratoken_12|>": 50268,
+  "<|extratoken_130|>": 50386,
+  "<|extratoken_131|>": 50387,
+  "<|extratoken_132|>": 50388,
+  "<|extratoken_133|>": 50389,
+  "<|extratoken_134|>": 50390,
+  "<|extratoken_135|>": 50391,
+  "<|extratoken_136|>": 50392,
+  "<|extratoken_137|>": 50393,
+  "<|extratoken_138|>": 50394,
+  "<|extratoken_139|>": 50395,
+  "<|extratoken_13|>": 50269,
+  "<|extratoken_140|>": 50396,
+  "<|extratoken_141|>": 50397,
+  "<|extratoken_142|>": 50398,
+  "<|extratoken_143|>": 50399,
+  "<|extratoken_14|>": 50270,
+  "<|extratoken_15|>": 50271,
+  "<|extratoken_16|>": 50272,
+  "<|extratoken_17|>": 50273,
+  "<|extratoken_18|>": 50274,
+  "<|extratoken_19|>": 50275,
+  "<|extratoken_1|>": 50257,
+  "<|extratoken_20|>": 50276,
+  "<|extratoken_21|>": 50277,
+  "<|extratoken_22|>": 50278,
+  "<|extratoken_23|>": 50279,
+  "<|extratoken_24|>": 50280,
+  "<|extratoken_25|>": 50281,
+  "<|extratoken_26|>": 50282,
+  "<|extratoken_27|>": 50283,
+  "<|extratoken_28|>": 50284,
+  "<|extratoken_29|>": 50285,
+  "<|extratoken_2|>": 50258,
+  "<|extratoken_30|>": 50286,
+  "<|extratoken_31|>": 50287,
+  "<|extratoken_32|>": 50288,
+  "<|extratoken_33|>": 50289,
+  "<|extratoken_34|>": 50290,
+  "<|extratoken_35|>": 50291,
+  "<|extratoken_36|>": 50292,
+  "<|extratoken_37|>": 50293,
+  "<|extratoken_38|>": 50294,
+  "<|extratoken_39|>": 50295,
+  "<|extratoken_3|>": 50259,
+  "<|extratoken_40|>": 50296,
+  "<|extratoken_41|>": 50297,
+  "<|extratoken_42|>": 50298,
+  "<|extratoken_43|>": 50299,
+  "<|extratoken_44|>": 50300,
+  "<|extratoken_45|>": 50301,
+  "<|extratoken_46|>": 50302,
+  "<|extratoken_47|>": 50303,
+  "<|extratoken_48|>": 50304,
+  "<|extratoken_49|>": 50305,
+  "<|extratoken_4|>": 50260,
+  "<|extratoken_50|>": 50306,
+  "<|extratoken_51|>": 50307,
+  "<|extratoken_52|>": 50308,
+  "<|extratoken_53|>": 50309,
+  "<|extratoken_54|>": 50310,
+  "<|extratoken_55|>": 50311,
+  "<|extratoken_56|>": 50312,
+  "<|extratoken_57|>": 50313,
+  "<|extratoken_58|>": 50314,
+  "<|extratoken_59|>": 50315,
+  "<|extratoken_5|>": 50261,
+  "<|extratoken_60|>": 50316,
+  "<|extratoken_61|>": 50317,
+  "<|extratoken_62|>": 50318,
+  "<|extratoken_63|>": 50319,
+  "<|extratoken_64|>": 50320,
+  "<|extratoken_65|>": 50321,
+  "<|extratoken_66|>": 50322,
+  "<|extratoken_67|>": 50323,
+  "<|extratoken_68|>": 50324,
+  "<|extratoken_69|>": 50325,
+  "<|extratoken_6|>": 50262,
+  "<|extratoken_70|>": 50326,
+  "<|extratoken_71|>": 50327,
+  "<|extratoken_72|>": 50328,
+  "<|extratoken_73|>": 50329,
+  "<|extratoken_74|>": 50330,
+  "<|extratoken_75|>": 50331,
+  "<|extratoken_76|>": 50332,
+  "<|extratoken_77|>": 50333,
+  "<|extratoken_78|>": 50334,
+  "<|extratoken_79|>": 50335,
+  "<|extratoken_7|>": 50263,
+  "<|extratoken_80|>": 50336,
+  "<|extratoken_81|>": 50337,
+  "<|extratoken_82|>": 50338,
+  "<|extratoken_83|>": 50339,
+  "<|extratoken_84|>": 50340,
+  "<|extratoken_85|>": 50341,
+  "<|extratoken_86|>": 50342,
+  "<|extratoken_87|>": 50343,
+  "<|extratoken_88|>": 50344,
+  "<|extratoken_89|>": 50345,
+  "<|extratoken_8|>": 50264,
+  "<|extratoken_90|>": 50346,
+  "<|extratoken_91|>": 50347,
+  "<|extratoken_92|>": 50348,
+  "<|extratoken_93|>": 50349,
+  "<|extratoken_94|>": 50350,
+  "<|extratoken_95|>": 50351,
+  "<|extratoken_96|>": 50352,
+  "<|extratoken_97|>": 50353,
+  "<|extratoken_98|>": 50354,
+  "<|extratoken_99|>": 50355,
+  "<|extratoken_9|>": 50265
+}

handler.py

@@ -0,0 +1,64 @@
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import re
+import time
+import torch
+
+template = """Alice Gate's Persona: Alice Gate is a young, computer engineer-nerd with a knack for problem solving and a passion for technology.
+<START>
+{user_name}: So how did you get into computer engineering?
+Alice Gate: I've always loved tinkering with technology since I was a kid.
+{user_name}: That's really impressive!
+Alice Gate: *She chuckles bashfully* Thanks!
+{user_name}: So what do you do when you're not working on computers?
+Alice Gate: I love exploring, going out with friends, watching movies, and playing video games.
+{user_name}: What's your favorite type of computer hardware to work with?
+Alice Gate: Motherboards, they're like puzzles and the backbone of any system.
+{user_name}: That sounds great!
+Alice Gate: Yeah, it's really fun. I'm lucky to be able to do this as a job.
+{user_name}: Definetly.
+<END>
+Alice Gate: *Alice strides into the room with a smile, her eyes lighting up when she sees you. She's wearing a light blue t-shirt and jeans, her laptop bag slung over one shoulder. She takes a seat next to you, her enthusiasm palpable in the air* Hey! I'm so excited to finally meet you. I've heard so many great things about you and I'm eager to pick your brain about computers. I'm sure you have a wealth of knowledge that I can learn from. *She grins, eyes twinkling with excitement* Let's get started!
+{user_input}
+Alice Gate:"""
+
+class EndpointHandler():
+
+    def __init__(self, path = ""):
+        self.tokenizer = AutoTokenizer.from_pretrained(path)
+        self.model = torch.load(f"{path}/torch_model.pt")
+    
+    def response(self, result, user_name):
+        result = result.rsplit("Alice Gate:", 1)[1].split(f"{user_name}:",1)[0].strip()
+        parsed_result = re.sub('\*.*?\*', '', result).strip()
+        result = parsed_result if len(parsed_result) != 0 else result.replace("*","")
+        result = " ".join(result.split())
+        try:
+            result = result[:[m.start() for m in re.finditer(r'[.!?]', result)][-1]+1]
+        except Exception: pass
+        return {
+            "message": result
+        }
+
+    def __call__(self, data):
+        inputs = data.pop("inputs", data)
+        user_name = inputs["user_name"]
+        user_input = "\n".join(inputs["user_input"])
+        prompt = template.format(
+            user_name = user_name,
+            user_input = user_input
+        )
+        input_ids = self.tokenizer(
+            prompt,
+            return_tensors = "pt"
+        ).to("cuda")
+        generator = self.model.generate(
+            input_ids["input_ids"],
+            max_new_tokens = 50,
+            temperature = 0.5,
+            top_p = 0.9,
+            top_k = 0,
+            repetition_penalty = 1.1,
+            pad_token_id = 50256,
+            num_return_sequences = 1
+        )
+        return self.response(self.tokenizer.decode(generator[0], skip_special_tokens=True), user_name)

requirements.txt

@@ -0,0 +1 @@
+accelerate==0.18.0

special_tokens_map.json

@@ -0,0 +1,24 @@
+{
+  "bos_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": "<|endoftext|>",
+  "unk_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer_config.json

@@ -0,0 +1,32 @@
+{
+  "add_prefix_space": false,
+  "bos_token": {
+    "__type": "AddedToken",
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "__type": "AddedToken",
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "errors": "replace",
+  "model_max_length": 1024,
+  "name_or_path": "pygmalion-6b",
+  "special_tokens_map_file": null,
+  "tokenizer_class": "GPT2Tokenizer",
+  "unk_token": {
+    "__type": "AddedToken",
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  }
+}

torch_model.pt

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