Nafiul Huda commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,7 +34,7 @@ import numpy as np
|
|
| 34 |
# 1. 全局配置与模型缓存管理
|
| 35 |
# ==========================================
|
| 36 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 37 |
-
device = "
|
| 38 |
dtype = torch.bfloat16
|
| 39 |
|
| 40 |
# 加载 safety checker（启动时加载一次）
|
|
|
|
| 34 |
# 1. 全局配置与模型缓存管理
|
| 35 |
# ==========================================
|
| 36 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 37 |
+
device = "cpu"
|
| 38 |
dtype = torch.bfloat16
|
| 39 |
|
| 40 |
# 加载 safety checker（启动时加载一次）
|