English
Fishfishfishfishfish commited on
Commit
3595599
·
verified ·
1 Parent(s): 3f77f51

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -17,10 +17,14 @@ its not very accurate yet, its trained on only 1.2mb of text
17
  Each safetensors file represents a different hidden dim value.
18
  Each trained for 1 epoch.
19
 
20
- inference.py must be edited for each safetensors.
21
 
22
  >sequence_length = 64
 
23
  >batch_size = 16
 
24
  >learning_rate = 0.0001
 
25
  >embedding_dim = 256
 
26
  >num_layers = 4
 
17
  Each safetensors file represents a different hidden dim value.
18
  Each trained for 1 epoch.
19
 
20
+ inference.py hidden dim value must be edited for each safetensors.
21
 
22
  >sequence_length = 64
23
+ >
24
  >batch_size = 16
25
+ >
26
  >learning_rate = 0.0001
27
+ >
28
  >embedding_dim = 256
29
+ >
30
  >num_layers = 4