Spaces:
Sleeping
Sleeping
Auto commit at 06-2025-08 23:47:33
Browse files
app.py
CHANGED
|
@@ -55,8 +55,7 @@ try:
|
|
| 55 |
token=HF_TOKEN,
|
| 56 |
torch_dtype=torch.float16,
|
| 57 |
trust_remote_code=True,
|
| 58 |
-
device_map="auto",
|
| 59 |
-
low_cpu_mem_usage=True
|
| 60 |
)
|
| 61 |
print(" ✅ 커스텀 모델 로딩 완료")
|
| 62 |
else:
|
|
|
|
| 55 |
token=HF_TOKEN,
|
| 56 |
torch_dtype=torch.float16,
|
| 57 |
trust_remote_code=True,
|
| 58 |
+
device_map="auto",
|
|
|
|
| 59 |
)
|
| 60 |
print(" ✅ 커스텀 모델 로딩 완료")
|
| 61 |
else:
|