Spaces:

MohamedRashad
/

Arabic-Nougat

Running on Zero

MohamedRashad commited on Aug 8

Commit

309d5b9

verified ·

1 Parent(s): e8d558d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -37,14 +37,14 @@ models_supported = {
             attn_implementation={"decoder": "flash_attention_2", "encoder": "eager"},
         ),
     ],
-    "mobser-small-v0.1": [
-        NougatProcessor.from_pretrained("TawasulAI/Mobser-Small-V0.1"),
-        VisionEncoderDecoderModel.from_pretrained(
-            "TawasulAI/Mobser-Small-V0.1",
-            torch_dtype=torch.bfloat16,
-            attn_implementation={"decoder": "flash_attention_2", "encoder": "eager"},
-        ),
-    ],
 }

             attn_implementation={"decoder": "flash_attention_2", "encoder": "eager"},
         ),
     ],
+    # "mobser-small-v0.1": [
+    #     NougatProcessor.from_pretrained("TawasulAI/Mobser-Small-V0.1"),
+    #     VisionEncoderDecoderModel.from_pretrained(
+    #         "TawasulAI/Mobser-Small-V0.1",
+    #         torch_dtype=torch.bfloat16,
+    #         attn_implementation={"decoder": "flash_attention_2", "encoder": "eager"},
+    #     ),
+    # ],
 }