Golali commited on
Commit
281e764
1 Parent(s): c9ac300

Create hubconf.yml

Browse files
Files changed (1) hide show
  1. hubconf.yml +17 -0
hubconf.yml ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ - id: your-model-name
3
+ name: Your Model Name
4
+ description: A brief description of your model.
5
+ model_file: path/to/your/model.pth
6
+ author: Your Name
7
+ language: English
8
+ license: MIT
9
+ tags:
10
+ - tag1
11
+ - tag2
12
+ pipelines:
13
+ - task: text-generation
14
+ source: prompt-based
15
+ architecture: GPT2
16
+ tokenizer: GPT2Tokenizer
17
+ model: path/to/your/model.pth