prithivMLmods commited on
Commit
1c5aa92
·
verified ·
1 Parent(s): 665e0de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -166,7 +166,7 @@ MODEL_ID_S = 'deepseek-ai/DeepSeek-OCR'
166
  processor_s = AutoProcessor.from_pretrained(MODEL_ID_S, trust_remote_code=True)
167
  model_s = AutoModelForCausalLM.from_pretrained(
168
  MODEL_ID_S,
169
- _attn_implementation='flash_attention_2',
170
  torch_dtype=torch.bfloat16,
171
  trust_remote_code=True,
172
  ).eval()
 
166
  processor_s = AutoProcessor.from_pretrained(MODEL_ID_S, trust_remote_code=True)
167
  model_s = AutoModelForCausalLM.from_pretrained(
168
  MODEL_ID_S,
169
+ _attn_implementation='eager',
170
  torch_dtype=torch.bfloat16,
171
  trust_remote_code=True,
172
  ).eval()