Upload GPT124MTextGenerationPipeline

config.json

@@ -7,6 +7,21 @@
     "AutoModelForCausalLM": "modeling_gpt.GPTModelForTextGeneration"
   },
   "block_size": 1024,
+  "custom_pipelines": {
+    "text-generation": {
+      "default": {
+        "model": {
+          "pt": "samkeet/GPT_124M"
+        }
+      },
+      "impl": "pipeline_gpt.GPT124MTextGenerationPipeline",
+      "pt": [
+        "AutoModelForCausalLM"
+      ],
+      "tf": [],
+      "type": "text"
+    }
+  },
   "model_type": "custom_gpt",
   "n_embd": 768,
   "n_head": 12,

pipeline_gpt.py

@@ -54,5 +54,5 @@ class GPT124MTextGenerationPipeline(Pipeline):
         """
         Decodes token ID into human-readable text using the tokenizer.
         """
-        print(model_output)
+
         return self.tokenizer.decode(model_output)