gbrabbit commited on
Commit
ea9b321
·
1 Parent(s): 0f0528f

Auto commit at 07-2025-08 0:24:17

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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="auto",
59
- low_cpu_mem_usage=False
 
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: