Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
from transformers import AutoTokenizer
|
2 |
from petals import AutoDistributedModelForCausalLM
|
3 |
-
import streamlit as st
|
4 |
# Choose any model available at https://health.petals.dev
|
5 |
model_name = "petals-team/StableBeluga2" # This one is fine-tuned Llama 2 (70B)
|
6 |
|
|
|
1 |
from transformers import AutoTokenizer
|
2 |
from petals import AutoDistributedModelForCausalLM
|
3 |
+
import streamlit as st
|
4 |
# Choose any model available at https://health.petals.dev
|
5 |
model_name = "petals-team/StableBeluga2" # This one is fine-tuned Llama 2 (70B)
|
6 |
|