Spaces:
Running
on
Zero
Running
on
Zero
del /data-nvme/zerogpu-offload/
Browse files
app.py
CHANGED
|
@@ -4,6 +4,9 @@ import gradio as gr
|
|
| 4 |
import torch
|
| 5 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 6 |
from huggingface_hub import whoami
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
# 定义系统提示语
|
| 9 |
system_prompt = """你是 Skywork-o1，Skywork AI 开发的思维模型，擅长通过深度思考解决涉及数学、编码和逻辑推理的复杂问题。面对用户请求时，你首先会进行一段漫长而深入的思考过程，探索问题的可能解决方案。完成思考后，你会在回复中详细解释解决过程。"""
|
|
|
|
| 4 |
import torch
|
| 5 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 6 |
from huggingface_hub import whoami
|
| 7 |
+
import os
|
| 8 |
+
|
| 9 |
+
os.removedirs('/data-nvme/zerogpu-offload/')
|
| 10 |
|
| 11 |
# 定义系统提示语
|
| 12 |
system_prompt = """你是 Skywork-o1，Skywork AI 开发的思维模型，擅长通过深度思考解决涉及数学、编码和逻辑推理的复杂问题。面对用户请求时，你首先会进行一段漫长而深入的思考过程，探索问题的可能解决方案。完成思考后，你会在回复中详细解释解决过程。"""
|