Upload folder using huggingface_hub

README.md

@@ -15,6 +15,7 @@ widget:
   - text: "Artist: Beyonce\nLyrics:"
   - text: "Artist: Drake\nLyrics:" 
 inference: true
+library_name: transformers
 ---
 
 # GPT-Neo 2.7B LoRA Fine-tuned for Lyrics Generation

inference_config.json

@@ -0,0 +1,16 @@
+{
+  "base_model": "EleutherAI/gpt-neo-2.7B",
+  "task": "text-generation",
+  "default_params": {
+    "max_length": 200,
+    "temperature": 0.7,
+    "top_p": 0.9,
+    "top_k": 50,
+    "do_sample": true,
+    "return_full_text": false
+  },
+  "prompt_format": "Artist: {artist}\nLyrics:",
+  "example_inputs": {
+    "artist": ["Taylor Swift", "Beyonce", "Drake", "Ed Sheeran", "Adele"]
+  }
+} 

library_config.json

@@ -0,0 +1,10 @@
+{
+  "architectures": ["GPTNeoForCausalLM"],
+  "model_type": "gpt_neo",
+  "library_name": "transformers",
+  "peft_type": "LORA",
+  "auto_map": {
+    "AutoModelForCausalLM": ["transformers", "GPTNeoForCausalLM"],
+    "AutoTokenizer": ["transformers", "GPT2Tokenizer"]
+  }
+}