Upload folder using huggingface_hub
Browse files- .gitattributes +4 -0
- README.md +102 -0
- WSB-GPT-7B_Q2_K.gguf +3 -0
- WSB-GPT-7B_Q4_K_M.gguf +3 -0
- WSB-GPT-7B_Q5_K_M.gguf +3 -0
- WSB-GPT-7B_Q8_0.gguf +3 -0
.gitattributes
CHANGED
@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
+
WSB-GPT-7B_Q2_K.gguf filter=lfs diff=lfs merge=lfs -text
|
37 |
+
WSB-GPT-7B_Q4_K_M.gguf filter=lfs diff=lfs merge=lfs -text
|
38 |
+
WSB-GPT-7B_Q5_K_M.gguf filter=lfs diff=lfs merge=lfs -text
|
39 |
+
WSB-GPT-7B_Q8_0.gguf filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
@@ -0,0 +1,102 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
datasets:
|
4 |
+
- Sentdex/wsb_reddit_v002
|
5 |
+
---
|
6 |
+
|
7 |
+
# Model Card for WSB-GPT-7B
|
8 |
+
|
9 |
+
This is a Llama 2 7B Chat model fine-tuned with QLoRA on 2017-2018ish /r/wallstreetbets subreddit comments and responses, with the hopes of learning more about QLoRA and creating models with a little more character.
|
10 |
+
|
11 |
+
|
12 |
+
### Model Description
|
13 |
+
|
14 |
+
- **Developed by:** Sentdex
|
15 |
+
- **Shared by:** Sentdex
|
16 |
+
- **GPU Compute provided by:** [Lambda Labs](https://lambdalabs.com/service/gpu-cloud)
|
17 |
+
|
18 |
+
- **Model type:** Instruct/Chat
|
19 |
+
- **Language(s) (NLP):** Multilingual from Llama 2, but not sure what the fine-tune did to it, or if the fine-tuned behavior translates well to other languages. Let me know!
|
20 |
+
- **License:** Apache 2.0
|
21 |
+
- **Finetuned from Llama 2 7B Chat**
|
22 |
+
|
23 |
+
|
24 |
+
- **Demo [optional]:** [More Information Needed]
|
25 |
+
|
26 |
+
## Uses
|
27 |
+
|
28 |
+
This model's primary purpose is to be a fun chatbot and to learn more about QLoRA. It is not intended to be used for any other purpose and some people may find it abrasive/offensive.
|
29 |
+
|
30 |
+
## Bias, Risks, and Limitations
|
31 |
+
|
32 |
+
This model is prone to using at least 3 words that were popularly used in the WSB subreddit in that era that are much more frowned-upon. As time goes on, I may wind up pruning or find-replacing these words in the training data, or leaving it.
|
33 |
+
|
34 |
+
Just be advised this model can be offensive and is not intended for all audiences!
|
35 |
+
|
36 |
+
## How to Get Started with the Model
|
37 |
+
### Prompt Format:
|
38 |
+
|
39 |
+
```
|
40 |
+
### Comment:
|
41 |
+
[parent comment text]
|
42 |
+
|
43 |
+
### REPLY:
|
44 |
+
[bot's reply]
|
45 |
+
|
46 |
+
### END.
|
47 |
+
```
|
48 |
+
|
49 |
+
Use the code below to get started with the model.
|
50 |
+
|
51 |
+
```py
|
52 |
+
from transformers import pipeline
|
53 |
+
|
54 |
+
# Initialize the pipeline for text generation using the Sentdex/WSB-GPT-7B model
|
55 |
+
pipe = pipeline("text-generation", model="Sentdex/WSB-GPT-7B")
|
56 |
+
|
57 |
+
# Define your prompt
|
58 |
+
prompt = """### Comment:
|
59 |
+
How does the stock market actually work?
|
60 |
+
|
61 |
+
### REPLY:
|
62 |
+
"""
|
63 |
+
|
64 |
+
# Generate text based on the prompt
|
65 |
+
generated_text = pipe(prompt, max_length=128, num_return_sequences=1)
|
66 |
+
|
67 |
+
# Extract and print the generated text
|
68 |
+
print(generated_text[0]['generated_text'].split("### END.")[0])
|
69 |
+
```
|
70 |
+
|
71 |
+
Example continued generation from above:
|
72 |
+
|
73 |
+
```
|
74 |
+
### Comment:
|
75 |
+
How does the stock market actually work?
|
76 |
+
|
77 |
+
### REPLY:
|
78 |
+
You sell when you are up and buy when you are down.
|
79 |
+
```
|
80 |
+
|
81 |
+
Despite `</s>` being the typical Llama stop token, I was never able to get this token to be generated in training/testing so the model would just never stop generating. I wound up testing with ### END. and that worked, but obviously isn't ideal. Will fix this in the future maybe(tm).
|
82 |
+
|
83 |
+
#### Hardware
|
84 |
+
|
85 |
+
This QLoRA was trained on a Lambda Labs 1x H100 80GB GPU instance.
|
86 |
+
|
87 |
+
## Citation
|
88 |
+
|
89 |
+
- Llama 2 (Meta AI) for the base model.
|
90 |
+
- Farouk E / Far El: https://twitter.com/far__el for helping with all my silly questions about QLoRA
|
91 |
+
- Lambda Labs for the compute. The model itself only took a few hours to train, but it took me days to learn how to tie everything together.
|
92 |
+
- Tim Dettmers, Artidoro Pagnoni, Ari Holtzman, Luke Zettlemoyer for QLoRA + implementation on github: https://github.com/artidoro/qlora/
|
93 |
+
- @eugene-yh and @jinyongyoo on Github + @ChrisHayduk for the QLoRA merge: https://gist.github.com/ChrisHayduk/1a53463331f52dca205e55982baf9930
|
94 |
+
|
95 |
+
|
96 |
+
## Model Card Contact
|
97 |
+
|
98 | |
99 |
+
|
100 |
+
***
|
101 |
+
|
102 |
+
Vanilla Quantization by [nold](https://huggingface.co/nold), Model by [WSB-GPT-7B](https://huggingface.co/Sentdex/WSB-GPT-7B)
|
WSB-GPT-7B_Q2_K.gguf
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:235bfebe62f834cdc64a77e88737471d8a47b9434cc55bd1847e1d49231e4c15
|
3 |
+
size 2532863712
|
WSB-GPT-7B_Q4_K_M.gguf
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:856eaff89db86fc319ede0eea6f8d02d5a9ffee089239e81c6ce69df840aa072
|
3 |
+
size 4081004256
|
WSB-GPT-7B_Q5_K_M.gguf
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c9b1c6c2a7c1727cf680ce3aba57c2faef3a129b97801f0304f5e83859077422
|
3 |
+
size 4783156960
|
WSB-GPT-7B_Q8_0.gguf
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:27ac9116c8bd5f7eb066c14ee58b6ee8a77fcca2e189b5a60d783561dea85811
|
3 |
+
size 7161089760
|