Spaces:

Geraldine
/

Image-to-text-SmolVLM-for-Omeka

Running

Geraldine commited on Feb 12

Commit

df7325b

verified ·

1 Parent(s): 35a385f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ import os, stat, io
 model = AutoModelForVision2Seq.from_pretrained(
     "./SmolVLM-500M-Instruct",
     torch_dtype=torch.float32,
-    #_attn_implementation="eager",
     device_map="cpu"
 )
 processor = AutoProcessor.from_pretrained("./SmolVLM-500M-Instruct")

 model = AutoModelForVision2Seq.from_pretrained(
     "./SmolVLM-500M-Instruct",
     torch_dtype=torch.float32,
+    _attn_implementation="eager",
     device_map="cpu"
 )
 processor = AutoProcessor.from_pretrained("./SmolVLM-500M-Instruct")