RinInori commited on
Commit
99ca5db
1 Parent(s): f4ad994

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
  from transformers import GenerationConfig, LlamaForCausalLM, LlamaTokenizer
6
  from transformers import Trainer
7
 
8
- BASE_MODEL = "TheBloke/stable-vicuna-13B-HF"
9
 
10
  # Create a custom device map
11
  # This will vary based on the architecture of model and the memory capacity of GPU and CPU
@@ -54,6 +54,6 @@ iface = gr.Interface(
54
  inputs="text",
55
  outputs="text",
56
  title="Chatbot",
57
- description="This vicuna app is using this model: https://huggingface.co/TheBloke/stable-vicuna-13B-HF"
58
  )
59
  iface.launch()
 
5
  from transformers import GenerationConfig, LlamaForCausalLM, LlamaTokenizer
6
  from transformers import Trainer
7
 
8
+ BASE_MODEL = "TheBloke/vicuna-7B-1.1-HF"
9
 
10
  # Create a custom device map
11
  # This will vary based on the architecture of model and the memory capacity of GPU and CPU
 
54
  inputs="text",
55
  outputs="text",
56
  title="Chatbot",
57
+ description="This vicuna app is using this model: https://huggingface.co/TheBloke/vicuna-7B-1.1-HF"
58
  )
59
  iface.launch()