Pravesh390 commited on
Commit
d812034
·
verified ·
1 Parent(s): c71b9dd

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +11 -50
README.md CHANGED
@@ -1,53 +1,14 @@
1
- ---
2
- library_name: transformers
3
- license: apache-2.0
4
- base_model: distilgpt2
5
- tags:
6
- - generated_from_trainer
7
- model-index:
8
- - name: distilgpt2-custom
9
- results: []
10
- ---
11
 
12
- <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
- should probably proofread and complete it, then remove this comment. -->
14
 
15
- # distilgpt2-custom
 
 
 
 
 
16
 
17
- This model is a fine-tuned version of [distilgpt2](https://huggingface.co/distilgpt2) on the None dataset.
18
-
19
- ## Model description
20
-
21
- More information needed
22
-
23
- ## Intended uses & limitations
24
-
25
- More information needed
26
-
27
- ## Training and evaluation data
28
-
29
- More information needed
30
-
31
- ## Training procedure
32
-
33
- ### Training hyperparameters
34
-
35
- The following hyperparameters were used during training:
36
- - learning_rate: 5e-05
37
- - train_batch_size: 2
38
- - eval_batch_size: 8
39
- - seed: 42
40
- - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
41
- - lr_scheduler_type: linear
42
- - num_epochs: 3
43
-
44
- ### Training results
45
-
46
-
47
-
48
- ### Framework versions
49
-
50
- - Transformers 4.53.2
51
- - Pytorch 2.6.0+cu124
52
- - Datasets 4.0.0
53
- - Tokenizers 0.21.2
 
1
+ # 📚 Pravesh390's Fine-tuned DistilGPT2
 
 
 
 
 
 
 
 
 
2
 
3
+ This model is fine-tuned on a small custom dataset with creative text samples.
4
+ Based on the [distilgpt2](https://huggingface.co/distilgpt2) model.
5
 
6
+ ## ✨ Usage
7
+ ```python
8
+ from transformers import pipeline
9
+ generator = pipeline("text-generation", model="Pravesh390/distilgpt2-custom")
10
+ generator("Once upon a time", max_length=50)
11
+ ```
12
 
13
+ ## 🛠️ Fine-tuned by:
14
+ [Pravesh390](https://huggingface.co/Pravesh390)