Spaces:
Sleeping
Sleeping
Auto commit at 07-2025-08 0:24:17
Browse files
app.py
CHANGED
|
@@ -55,8 +55,9 @@ try:
|
|
| 55 |
token=HF_TOKEN,
|
| 56 |
torch_dtype=torch.float16,
|
| 57 |
trust_remote_code=True,
|
| 58 |
-
device_map=
|
| 59 |
-
low_cpu_mem_usage=
|
|
|
|
| 60 |
)
|
| 61 |
print(" ✅ 커스텀 모델 로딩 완료")
|
| 62 |
else:
|
|
|
|
| 55 |
token=HF_TOKEN,
|
| 56 |
torch_dtype=torch.float16,
|
| 57 |
trust_remote_code=True,
|
| 58 |
+
device_map=None,
|
| 59 |
+
low_cpu_mem_usage=True,
|
| 60 |
+
# max_memory={0: "4GB"} # GPU 메모리 제한
|
| 61 |
)
|
| 62 |
print(" ✅ 커스텀 모델 로딩 완료")
|
| 63 |
else:
|