Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -39,7 +39,8 @@ LANGUAGES = {
|
|
| 39 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 40 |
|
| 41 |
device = "cuda"
|
| 42 |
-
MODEL_NAME = "ByteDance-Seed/Seed-X-PPO-7B"
|
|
|
|
| 43 |
print("Start dowload")
|
| 44 |
def load_model():
|
| 45 |
model = AutoModelForCausalLM.from_pretrained(MODEL_NAME,torch_dtype="bfloat16").to(device)
|
|
@@ -151,9 +152,9 @@ with gr.Blocks(title="Seed-X") as demo:
|
|
| 151 |
outputs=target_text
|
| 152 |
)
|
| 153 |
|
| 154 |
-
gr.Markdown(
|
| 155 |
-
|
| 156 |
-
)
|
| 157 |
|
| 158 |
# 启动应用
|
| 159 |
if __name__ == "__main__":
|
|
|
|
| 39 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 40 |
|
| 41 |
device = "cuda"
|
| 42 |
+
# MODEL_NAME = "ByteDance-Seed/Seed-X-PPO-7B"
|
| 43 |
+
MODEL_NAME="Qwen/Qwen3-Next-80B-A3B-Instruct"
|
| 44 |
print("Start dowload")
|
| 45 |
def load_model():
|
| 46 |
model = AutoModelForCausalLM.from_pretrained(MODEL_NAME,torch_dtype="bfloat16").to(device)
|
|
|
|
| 152 |
outputs=target_text
|
| 153 |
)
|
| 154 |
|
| 155 |
+
# gr.Markdown(
|
| 156 |
+
# "🌐[Github](https://github.com/ByteDance-Seed/Seed-X-7B) 📄[Report](https://arxiv.org/pdf/2507.13618) 🤗[Model](https://huggingface.co/collections/ByteDance-Seed/seed-x-6878753f2858bc17afa78543)"
|
| 157 |
+
# )
|
| 158 |
|
| 159 |
# 启动应用
|
| 160 |
if __name__ == "__main__":
|