Spaces:

numind
/

NuExtract-2.0

Sleeping

Alexandre-Numind commited on Jun 16

Commit

dc29570

verified ·

1 Parent(s): e77d9f4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,6 @@ model = AutoModelForVision2Seq.from_pretrained(
     model_name,
     trust_remote_code=True,
     torch_dtype=torch.bfloat16,
-    attn_implementation="flash_attention_2",
     device_map="auto",
 )
 processor = AutoProcessor.from_pretrained(

     model_name,
     trust_remote_code=True,
     torch_dtype=torch.bfloat16,
     device_map="auto",
 )
 processor = AutoProcessor.from_pretrained(