Update app.py

app.py

@@ -3,6 +3,12 @@ import os
 os.environ['CUDA_VISIBLE_DEVICES'] = ''
 os.environ['ZEROGPU'] = '1'  # ZeroGPU 환경임을 표시
 
+# Safetensors 사용 강제 및 torch 버전 체크 우회
+os.environ['SAFETENSORS_FAST_GPU'] = '1'
+os.environ['TRANSFORMERS_OFFLINE'] = '0'
+os.environ['TRANSFORMERS_USE_SAFETENSORS'] = '1'
+os.environ['DISABLE_TORCH_LOAD_SAFETY_CHECK'] = '1'  # This might help bypass the check
+
 import spaces  # spaces import는 환경 설정 후에
 import shlex
 import subprocess
@@ -10,6 +16,17 @@ import subprocess
 # 라이브러리 버전 호환성 문제 해결
 subprocess.run(shlex.split("pip install pip==24.0"), check=True)
 
+# Safetensors를 기본으로 사용하도록 transformers 설정
+subprocess.run(shlex.split("pip install safetensors --upgrade"), check=True)
+
+# Monkey patch to force safetensors loading
+import sys
+if 'transformers' in sys.modules:
+    del sys.modules['transformers']
+    
+# transformers import 전에 환경 변수 설정
+os.environ['TRANSFORMERS_USE_SAFETENSORS'] = '1'
+
 subprocess.run(
     shlex.split(
         "pip install package/onnxruntime_gpu-1.17.0-cp310-cp310-manylinux_2_28_x86_64.whl --force-reinstall --no-deps"