Update app.py

app.py

@@ -16,7 +16,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
 moondream = AutoModelForCausalLM.from_pretrained(
     "Yakshaving/moondream-paint", trust_remote_code=True,
     torch_dtype=torch.bfloat16, device_map={"": "cuda"},
-    attn_implementation="flash_attention_2"
+    # attn_implementation="flash_attention_2"
 )
 moondream.eval()