Spaces:
Running
Running
Commit
•
7293a15
1
Parent(s):
e25fb5a
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ import subprocess
|
|
17 |
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
18 |
|
19 |
models = {
|
20 |
-
'microsoft/Florence-2-base': AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True).to("
|
21 |
}
|
22 |
|
23 |
processors = {
|
@@ -44,7 +44,7 @@ def run_example(task_prompt, image, text_input=None, model_id='microsoft/Florenc
|
|
44 |
prompt = task_prompt
|
45 |
else:
|
46 |
prompt = task_prompt + text_input
|
47 |
-
inputs = processor(text=prompt, images=image, return_tensors="pt").to("
|
48 |
generated_ids = model.generate(
|
49 |
input_ids=inputs["input_ids"],
|
50 |
pixel_values=inputs["pixel_values"],
|
|
|
17 |
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
18 |
|
19 |
models = {
|
20 |
+
'microsoft/Florence-2-base': AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True).to("cpu").eval(),
|
21 |
}
|
22 |
|
23 |
processors = {
|
|
|
44 |
prompt = task_prompt
|
45 |
else:
|
46 |
prompt = task_prompt + text_input
|
47 |
+
inputs = processor(text=prompt, images=image, return_tensors="pt").to("cpu")
|
48 |
generated_ids = model.generate(
|
49 |
input_ids=inputs["input_ids"],
|
50 |
pixel_values=inputs["pixel_values"],
|