liuyizhang commited on
Commit
0d72022
·
1 Parent(s): 13315d1
model/README.md DELETED
@@ -1,46 +0,0 @@
1
- ---
2
- license: mit
3
- tags:
4
- - generated_from_trainer
5
- model-index:
6
- - name: prompt-extend
7
- results: []
8
- ---
9
- [![Generic badge](https://img.shields.io/badge/🤗-Open%20in%20Spaces-blue.svg)](https://huggingface.co/spaces/daspartho/prompt-extend)
10
-
11
- # Prompt Extend
12
-
13
- Text generation model for generating suitable style cues given the main idea for a prompt.
14
-
15
- It is a GPT-2 model trained on [dataset](https://huggingface.co/datasets/daspartho/stable-diffusion-prompts) of stable diffusion prompts.
16
-
17
- ### Training hyperparameters
18
-
19
- The following hyperparameters were used during training:
20
- - learning_rate: 0.0001
21
- - train_batch_size: 128
22
- - eval_batch_size: 256
23
- - seed: 42
24
- - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
25
- - lr_scheduler_type: cosine
26
- - lr_scheduler_warmup_ratio: 0.1
27
- - num_epochs: 5
28
- - mixed_precision_training: Native AMP
29
-
30
- ### Training results
31
-
32
- | Training Loss | Epoch | Step | Validation Loss |
33
- |:-------------:|:-----:|:-----:|:---------------:|
34
- | 3.7436 | 1.0 | 12796 | 2.5429 |
35
- | 2.3292 | 2.0 | 25592 | 2.0711 |
36
- | 1.9439 | 3.0 | 38388 | 1.8447 |
37
- | 1.7059 | 4.0 | 51184 | 1.7325 |
38
- | 1.5775 | 5.0 | 63980 | 1.7110 |
39
-
40
-
41
- ### Framework versions
42
-
43
- - Transformers 4.24.0
44
- - Pytorch 1.13.0+cu117
45
- - Datasets 2.7.1
46
- - Tokenizers 0.13.2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
model/config.json DELETED
@@ -1,39 +0,0 @@
1
- {
2
- "_name_or_path": "gpt2",
3
- "activation_function": "gelu_new",
4
- "architectures": [
5
- "GPT2LMHeadModel"
6
- ],
7
- "attn_pdrop": 0.1,
8
- "bos_token_id": 0,
9
- "embd_pdrop": 0.1,
10
- "eos_token_id": 0,
11
- "initializer_range": 0.02,
12
- "layer_norm_epsilon": 1e-05,
13
- "model_type": "gpt2",
14
- "n_ctx": 128,
15
- "n_embd": 768,
16
- "n_head": 12,
17
- "n_inner": null,
18
- "n_layer": 12,
19
- "n_positions": 1024,
20
- "reorder_and_upcast_attn": false,
21
- "resid_pdrop": 0.1,
22
- "scale_attn_by_inverse_layer_idx": false,
23
- "scale_attn_weights": true,
24
- "summary_activation": null,
25
- "summary_first_dropout": 0.1,
26
- "summary_proj_to_labels": true,
27
- "summary_type": "cls_index",
28
- "summary_use_proj": true,
29
- "task_specific_params": {
30
- "text-generation": {
31
- "do_sample": true,
32
- "max_length": 50
33
- }
34
- },
35
- "torch_dtype": "float32",
36
- "transformers_version": "4.24.0",
37
- "use_cache": true,
38
- "vocab_size": 52000
39
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
model/merges.txt DELETED
The diff for this file is too large to render. See raw diff
 
model/pytorch_model.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:9b2b0c3fa1c9ca9c46fcde7690c72facfdc736802ec22d9144db4f82ae9a4f9e
3
- size 515752509
 
 
 
 
model/special_tokens_map.json DELETED
@@ -1,6 +0,0 @@
1
- {
2
- "bos_token": "<|endoftext|>",
3
- "eos_token": "<|endoftext|>",
4
- "pad_token": "<|endoftext|>",
5
- "unk_token": "<|endoftext|>"
6
- }
 
 
 
 
 
 
 
model/tokenizer.json DELETED
The diff for this file is too large to render. See raw diff
 
model/tokenizer_config.json DELETED
@@ -1,10 +0,0 @@
1
- {
2
- "add_prefix_space": false,
3
- "bos_token": "<|endoftext|>",
4
- "eos_token": "<|endoftext|>",
5
- "model_max_length": 1024,
6
- "name_or_path": "daspartho/prompt-tokenizer",
7
- "special_tokens_map_file": null,
8
- "tokenizer_class": "GPT2Tokenizer",
9
- "unk_token": "<|endoftext|>"
10
- }
 
 
 
 
 
 
 
 
 
 
 
model/training_args.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:feb1020a4f463c27cf23421abb9ec75003c289d820d4ebbe52b15d9af77b46c2
3
- size 3387
 
 
 
 
model/vocab.json DELETED
The diff for this file is too large to render. See raw diff