prithivMLmods commited on
Commit
3b016e9
·
verified ·
1 Parent(s): 13ed02d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -250,7 +250,6 @@ phi4_model = AutoModelForCausalLM.from_pretrained(
250
  device_map="auto",
251
  torch_dtype="auto",
252
  trust_remote_code=True,
253
- _attn_implementation="eager",
254
  )
255
 
256
  # ------------------------------------------------------------------------------
 
250
  device_map="auto",
251
  torch_dtype="auto",
252
  trust_remote_code=True,
 
253
  )
254
 
255
  # ------------------------------------------------------------------------------