enryu43 commited on
Commit
bd336d3
1 Parent(s): c9150ca

Initial commit

Browse files
Files changed (4) hide show
  1. README.md +2 -0
  2. config.json +33 -0
  3. pytorch_model.bin +3 -0
  4. tf_model.h5 +3 -0
README.md ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ Autoregressive prompt augmenter for https://medium.com/@enryu9000/TODO.
2
+
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_function": "gelu_new",
3
+ "architectures": [
4
+ "GPT2LMHeadModel"
5
+ ],
6
+ "attn_pdrop": 0.1,
7
+ "bos_token_id": 12221,
8
+ "embd_pdrop": 0.1,
9
+ "eos_token_id": 12222,
10
+ "initializer_range": 0.02,
11
+ "layer_norm_epsilon": 1e-05,
12
+ "model_type": "gpt2",
13
+ "n_embd": 768,
14
+ "n_head": 12,
15
+ "n_inner": 1536,
16
+ "n_layer": 12,
17
+ "n_positions": 52,
18
+ "name_or_path": "anifusion-sd-prompt-augmenter",
19
+ "pad_token_id": 12223,
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
+ "torch_dtype": "float32",
30
+ "transformers_version": "4.23.1",
31
+ "use_cache": true,
32
+ "vocab_size": 12224
33
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88362977ebe97bea20d2967877853fbbe9cac46a6dd3b842202444dc8a5deefd
3
+ size 264701501
tf_model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f93a25f8aa8a2821719d0682ea44f4f1d2a86e45ef1b7476c5de93c30a6b726b
3
+ size 264792168