Spaces:

MohamedRashad
/

Arabic-Nougat

Running on Zero

MohamedRashad commited on Aug 8

Commit

84ded95

verified ·

1 Parent(s): cd3b583

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ models_supported = {
         VisionEncoderDecoderModel.from_pretrained(
             "MohamedRashad/arabic-base-nougat",
             torch_dtype=torch.bfloat16,
-            attn_implementation={"decoder": "flash_attention_2", "encoder": "eager"},
         ),
     ],
     "arabic-large-nougat": [
@@ -34,7 +34,7 @@ models_supported = {
         VisionEncoderDecoderModel.from_pretrained(
             "MohamedRashad/arabic-large-nougat",
             torch_dtype=torch.bfloat16,
-            attn_implementation={"decoder": "flash_attention_2", "encoder": "eager"},
         ),
     ],
     # "mobser-small-v0.1": [

         VisionEncoderDecoderModel.from_pretrained(
             "MohamedRashad/arabic-base-nougat",
             torch_dtype=torch.bfloat16,
+            # attn_implementation={"decoder": "flash_attention_2", "encoder": "eager"},
         ),
     ],
     "arabic-large-nougat": [
         VisionEncoderDecoderModel.from_pretrained(
             "MohamedRashad/arabic-large-nougat",
             torch_dtype=torch.bfloat16,
+            # attn_implementation={"decoder": "flash_attention_2", "encoder": "eager"},
         ),
     ],
     # "mobser-small-v0.1": [