Spaces:

numind
/

NuExtract

Running on L4

Alexandre-Numind commited on Jun 20

Commit

8983ec1

•

1 Parent(s): 5e52f36

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -307,8 +307,6 @@ model = AutoModelForCausalLM.from_pretrained(
         "numind/NuExtract",
         trust_remote_code=True,
         device_map="cuda",
-        torch_dtype="auto",
-        attn_implementation="flash_attention_2"
         )

         "numind/NuExtract",
         trust_remote_code=True,
         device_map="cuda",
         )