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"
) |