File size: 595 Bytes
64910dd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
pipeline_config = {
    "task": "text-generation",
    "framework": "pt",
    "model": {
        "name": "ombhojane/wellness-mini",
        "type": "AutoModelForCausalLM"
    },
    "tokenizer": {
        "name": "ombhojane/wellness-mini",
        "type": "AutoTokenizer"
    }
}

# Save pipeline config
import json
with open("pipeline_config.json", "w") as f:
    json.dump(pipeline_config, f, indent=2)

# Upload pipeline config
api.upload_file(
    path_or_fileobj="pipeline_config.json",
    path_in_repo="pipeline_config.json",
    repo_id="ombhojane/wellness-mini",
    repo_type="model"
)