Florence-2-DocVQA

Running on Zero

gokaygokay commited on Jun 20

Commit

b9bfc9d

•

1 Parent(s): e62701c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ import numpy as np
 import subprocess
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-model_id = 'microsoft/Florence-2-large-ft'
 model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True).to("cuda").eval()
 processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)

 import subprocess
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+model_id = 'microsoft/Florence-2-large'
 model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True).to("cuda").eval()
 processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)