ldhldh commited on
Commit
942ae7e
·
1 Parent(s): d0a28db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -1,7 +1,11 @@
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"
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer
3
+ import os
4
+
5
+ os.system("pip install git+https://github.com/bigscience-workshop/petals")
6
+
7
  from petals import AutoDistributedModelForCausalLM
8
+ import npc_data
9
 
10
  # Choose any model available at https://health.petals.dev
11
  model_name = "daekeun-ml/Llama-2-ko-instruct-13B"