First commit

config/env_setup.py

@@ -0,0 +1,12 @@
+import os
+import torch
+
+def setup_environment():
+    """
+    OCR 모델 실행에 필요한 환경 변수를 설정합니다.
+    """
+    if torch.version.cuda == "11.8":
+        os.environ["TRITON_PTXAS_PATH"] = "/usr/local/cuda-11.8/bin/ptxas"
+
+    os.environ["VLLM_USE_V1"] = "0"
+    os.environ["CUDA_VISIBLE_DEVICES"] = "0"