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