Instructions to use difinative/AIBuddy with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use difinative/AIBuddy with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("difinative/AIBuddy") model = AutoModelForSeq2SeqLM.from_pretrained("difinative/AIBuddy", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit ·
f43c210
1
Parent(s): bbce628
Update README.md
Browse files
README.md
CHANGED
|
@@ -8,7 +8,7 @@ Model Details
|
|
| 8 |
Model: BART (facebook/bart-large)
|
| 9 |
Task: CLI Command Generation from Human Instructions
|
| 10 |
Tokenizer: BART Tokenizer (facebook/bart-large)
|
| 11 |
-
Fine-Tuning Hyperparameters: num_epochs=
|
| 12 |
|
| 13 |
Usage
|
| 14 |
You can use this model to generate CLI commands from human instructions. You can either directly use the model for inference or integrate it into applications using the provided Gradio interface.
|
|
|
|
| 8 |
Model: BART (facebook/bart-large)
|
| 9 |
Task: CLI Command Generation from Human Instructions
|
| 10 |
Tokenizer: BART Tokenizer (facebook/bart-large)
|
| 11 |
+
Fine-Tuning Hyperparameters: num_epochs=2, batch_size=4, accumulation_steps=2, learning_rate=2e-5
|
| 12 |
|
| 13 |
Usage
|
| 14 |
You can use this model to generate CLI commands from human instructions. You can either directly use the model for inference or integrate it into applications using the provided Gradio interface.
|