jekunz commited on
Commit
26b86d9
·
verified ·
1 Parent(s): 1cf485e

Training in progress, step 50000

Browse files
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: LumiOpen/Viking-7B
3
+ library_name: transformers
4
+ model_name: Viking-7B-smol-smoltalk-sv
5
+ tags:
6
+ - generated_from_trainer
7
+ - sft
8
+ - trl
9
+ licence: license
10
+ ---
11
+
12
+ # Model Card for Viking-7B-smol-smoltalk-sv
13
+
14
+ This model is a fine-tuned version of [LumiOpen/Viking-7B](https://huggingface.co/LumiOpen/Viking-7B).
15
+ It has been trained using [TRL](https://github.com/huggingface/trl).
16
+
17
+ ## Quick start
18
+
19
+ ```python
20
+ from transformers import pipeline
21
+
22
+ question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
23
+ generator = pipeline("text-generation", model="liu-nlp/Viking-7B-smol-smoltalk-sv", device="cuda")
24
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
25
+ print(output["generated_text"])
26
+ ```
27
+
28
+ ## Training procedure
29
+
30
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/jenny-kunz-liu/huggingface/runs/zvs1omc5)
31
+
32
+
33
+ This model was trained with SFT.
34
+
35
+ ### Framework versions
36
+
37
+ - TRL: 0.21.0
38
+ - Transformers: 4.55.1
39
+ - Pytorch: 2.8.0
40
+ - Datasets: 4.0.0
41
+ - Tokenizers: 0.21.4
42
+
43
+ ## Citations
44
+
45
+
46
+
47
+ Cite TRL as:
48
+
49
+ ```bibtex
50
+ @misc{vonwerra2022trl,
51
+ title = {{TRL: Transformer Reinforcement Learning}},
52
+ author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallou{\'e}dec},
53
+ year = 2020,
54
+ journal = {GitHub repository},
55
+ publisher = {GitHub},
56
+ howpublished = {\url{https://github.com/huggingface/trl}}
57
+ }
58
+ ```
config.json CHANGED
@@ -24,7 +24,7 @@
24
  "rope_theta": 10000.0,
25
  "tie_word_embeddings": false,
26
  "torch_dtype": "bfloat16",
27
- "transformers_version": "4.52.4",
28
  "untie_embeddings_and_output_weights": true,
29
  "use_cache": true,
30
  "vocab_size": 131072
 
24
  "rope_theta": 10000.0,
25
  "tie_word_embeddings": false,
26
  "torch_dtype": "bfloat16",
27
+ "transformers_version": "4.55.1",
28
  "untie_embeddings_and_output_weights": true,
29
  "use_cache": true,
30
  "vocab_size": 131072
model-00001-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9c3795263c77d3ff227ab335e6ca7b3e66b61c5f72085566c5b521a64a4283ec
3
  size 4941047672
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f6797634a480ceec712a996b54d2bd90f3fb2da53608356ab6e9811eb1d7da7a
3
  size 4941047672
model-00002-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2ee392ebb1bc30b581241a7fd227928bda1205d4950ee496066f3c177004e248
3
  size 4947390880
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e201d7b0ce91da145e3bdfde9ffd5f715ff84a1865172ad8d8bf0a05c1adf0f
3
  size 4947390880
model-00003-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8e63b0455ee0d632688da35f5d357e3b4e5273ad8185b097590287f2dbaa6376
3
  size 4137880232
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e52fe83c4ec89afb37bed5733294105569c6ffeea70007a65ae1605d48d66307
3
  size 4137880232
model-00004-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:baa88ce8c9a40ee950fc995ecb7425b7bfaf6cf1bb310f1bad5100ef20353e9f
3
  size 1073741952
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:951f7a781903b5c48e92025e6e29ce2ea5417e128bd45d04fbcdcf668de7f89c
3
  size 1073741952
model.safetensors.index.json CHANGED
@@ -1,5 +1,6 @@
1
  {
2
  "metadata": {
 
3
  "total_size": 15100026880
4
  },
5
  "weight_map": {
 
1
  {
2
  "metadata": {
3
+ "total_parameters": 7550013440,
4
  "total_size": 15100026880
5
  },
6
  "weight_map": {
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:51f51d4404aecb3d521c551a0276c6d9e2f3abb1502a8a55174efbc5ac811304
3
- size 5624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:51dad3a86f75c1ff687037b47a2fc8c012ac4d54c4995a14e56b8920cd21eb72
3
+ size 6161