Upload folder using huggingface_hub

README.md

@@ -33,7 +33,7 @@ You can use this model through its own custom pipeline:
 from transformers import pipeline
 
 pipe = pipeline(
-  task="gpt2-exon-introns-classification",
+  task="gpt2-exon-intron-classification",
   model="GustavoHCruz/ExInGPT",
   trust_remote_code=True,
 )

config.json

@@ -1,15 +1,19 @@
 {
   "activation_function": "gelu_new",
-  "architectures": [
-    "GPT2LMHeadModel"
-  ],
+  "architectures": ["GPT2LMHeadModel"],
   "attn_pdrop": 0.1,
   "bos_token_id": 50256,
+  "custom_pipelines": {
+    "gpt2-exon-intron-classification": {
+      "impl": "gpt2_exon_intron_classification.GPT2ExonIntronClassificationPipeline",
+      "pt": ["GPT2LMHeadModel"]
+    }
+  },
   "dtype": "float32",
   "embd_pdrop": 0.1,
   "eos_token_id": 50256,
   "initializer_range": 0.02,
-  "layer_norm_epsilon": 1e-05,
+  "layer_norm_epsilon": 1e-5,
   "model_type": "gpt2",
   "n_ctx": 1024,
   "n_embd": 768,

gpt2_exon_intron_classification.py

@@ -1,5 +1,3 @@
-
-
 from typing import Any, Optional
 
 import torch
@@ -179,4 +177,4 @@ PIPELINE_REGISTRY.register_pipeline(
 	"gpt2-exon-intron-classification",
 	pipeline_class=GPT2ExonIntronClassificationPipeline,
 	pt_model=GPT2LMHeadModel,
-)
+)