Hugging Face's logo

Spaces:
ldhldh
/
llm_server
Runtime error

App Files Community
ldhldh commited on
Commit
e00309c
·
1 Parent(s): c07b63c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer
3
  from petals import AutoDistributedModelForCausalLM
 
4
 
5
  # Choose any model available at https://health.petals.dev
6
  model_name = "daekeun-ml/Llama-2-ko-instruct-13B"
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer
3
  from petals import AutoDistributedModelForCausalLM
4
+ import data
5
 
6
  # Choose any model available at https://health.petals.dev
7
  model_name = "daekeun-ml/Llama-2-ko-instruct-13B"