Portx
/

trial-multi-model

Model card Files Files and versions

Portx commited on Mar 2, 2025

Commit

a2b6dcd

·

verified ·

1 Parent(s): 96ec3d9

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -3,8 +3,8 @@ import sys
 from subprocess import run
 from PIL import Image
-run("pip install flash-attn --no-build-isolation", shell=True, check=True)
-#run("pip install -U bitsandbytes", shell=True, check=True)
 from transformers import AutoModelForVision2Seq, AutoProcessor, BitsAndBytesConfig

 from subprocess import run
 from PIL import Image
+#run("pip install flash-attn --no-build-isolation", shell=True, check=True)
+run("pip install -U bitsandbytes", shell=True, check=True)
 from transformers import AutoModelForVision2Seq, AutoProcessor, BitsAndBytesConfig