Hugging Face's logo

Spaces:
johnbridges
/
TestHolo
Sleeping

App Files Community
johnbridges commited on
Commit
a727012
·
verified ·
1 Parent(s): 5ddb3ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -28,7 +28,6 @@ try:
28
  model = AutoModelForImageTextToText.from_pretrained(
29
  MODEL_ID,
30
  torch_dtype=torch.bfloat16,
31
- attn_implementation="flash_attention_2",
32
  trust_remote_code=True
33
  ).to("cuda")
34
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
 
28
  model = AutoModelForImageTextToText.from_pretrained(
29
  MODEL_ID,
30
  torch_dtype=torch.bfloat16,
 
31
  trust_remote_code=True
32
  ).to("cuda")
33
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)