Spaces:
Sleeping
Sleeping
Commit
·
5efa561
0
Parent(s):
Duplicate from johnwick123forevr/Testtrial1
Browse files- .gitattributes +31 -0
- README.md +14 -0
- app.py +53 -0
- requirements.txt +3 -0
.gitattributes
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 5 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
| 6 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
| 7 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 8 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 9 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
| 10 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
| 11 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
| 12 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
| 13 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
| 14 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 15 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
| 16 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
| 17 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
| 18 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
| 20 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 21 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
| 22 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
| 23 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
| 24 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
| 25 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
| 26 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
| 27 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
| 28 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
| 29 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 30 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 31 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
title: WizardLM
|
| 3 |
+
emoji: 😁
|
| 4 |
+
colorFrom: yellow
|
| 5 |
+
colorTo: green
|
| 6 |
+
sdk: gradio
|
| 7 |
+
sdk_version: 3.35.2
|
| 8 |
+
app_file: app.py
|
| 9 |
+
pinned: false
|
| 10 |
+
license: apache-2.0
|
| 11 |
+
duplicated_from: johnwick123forevr/Testtrial1
|
| 12 |
+
---
|
| 13 |
+
|
| 14 |
+
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
app.py
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import gradio as gr
|
| 2 |
+
import os
|
| 3 |
+
import wget
|
| 4 |
+
from llama_cpp import Llama
|
| 5 |
+
import random
|
| 6 |
+
url = 'https://huggingface.co/TheBloke/WizardLM-7B-uncensored-GGML/resolve/main/WizardLM-7B-uncensored.ggmlv3.q4_0.bin'
|
| 7 |
+
filename = wget.download(url)
|
| 8 |
+
llm2 = Llama(model_path=filename, seed=random.randint(1, 2**31))
|
| 9 |
+
theme = gr.themes.Soft(
|
| 10 |
+
primary_hue=gr.themes.Color("#ededed", "#fee2e2", "#fecaca", "#fca5a5", "#f87171", "#ef4444", "#dc2626", "#b91c1c", "#991b1b", "#7f1d1d", "#6c1e1e"),
|
| 11 |
+
neutral_hue="red",
|
| 12 |
+
)
|
| 13 |
+
title = """<h1 align="center">Chat with awesome WizardLM 7b model!</h1><br>"""
|
| 14 |
+
with gr.Blocks(theme=theme) as demo:
|
| 15 |
+
gr.HTML(title)
|
| 16 |
+
gr.HTML("This model is awesome for its size! It is only 20th the size of Chatgpt but is around 90% as good as Chatgpt. However, please don't rely on WizardLM to provide 100% true information as it might be wrong sometimes. ")
|
| 17 |
+
chatbot = gr.Chatbot()
|
| 18 |
+
msg = gr.Textbox()
|
| 19 |
+
clear = gr.ClearButton([msg, chatbot])
|
| 20 |
+
#instruction = gr.Textbox(label="Instruction", placeholder=)
|
| 21 |
+
|
| 22 |
+
def user(user_message, history):
|
| 23 |
+
return gr.update(value="", interactive=True), history + [[user_message, None]]
|
| 24 |
+
|
| 25 |
+
def bot(history):
|
| 26 |
+
#instruction = history[-1][1] or ""
|
| 27 |
+
user_message = history[-1][0]
|
| 28 |
+
#token1 = llm.tokenize(b"### Instruction: ")
|
| 29 |
+
#token2 = llm.tokenize(instruction.encode())
|
| 30 |
+
#token3 = llm2.tokenize(b"USER: ")
|
| 31 |
+
tokens3 = llm2.tokenize(user_message.encode())
|
| 32 |
+
token4 = llm2.tokenize(b"\n\n### Response:")
|
| 33 |
+
tokens = tokens3 + token4
|
| 34 |
+
history[-1][1] = ""
|
| 35 |
+
count = 0
|
| 36 |
+
output = ""
|
| 37 |
+
for token in llm2.generate(tokens, top_k=50, top_p=0.73, temp=0.72, repeat_penalty=1.1):
|
| 38 |
+
text = llm2.detokenize([token])
|
| 39 |
+
output += text.decode()
|
| 40 |
+
count += 1
|
| 41 |
+
if count >= 500 or (token == llm2.token_eos()):
|
| 42 |
+
break
|
| 43 |
+
history[-1][1] += text.decode()
|
| 44 |
+
yield history
|
| 45 |
+
|
| 46 |
+
response = msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
|
| 47 |
+
bot, chatbot, chatbot
|
| 48 |
+
)
|
| 49 |
+
response.then(lambda: gr.update(interactive=True), None, [msg], queue=False)
|
| 50 |
+
gr.HTML("Thanks for checking out this app!")
|
| 51 |
+
|
| 52 |
+
demo.queue()
|
| 53 |
+
demo.launch(debug=True)
|
requirements.txt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
llama-cpp-python
|
| 2 |
+
wget
|
| 3 |
+
gradio
|