Spaces:

HuggingFaceM4
/

idefics-8b

Running on Zero

merve HF staff commited on Apr 23

Commit

95e98bf

•

1 Parent(s): 7aa50ed

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ model = Idefics2ForConditionalGeneration.from_pretrained(
         #_attn_implementation="flash_attention_2",
         trust_remote_code=True).to("cuda")
 def model_inference(
     image, text, decoding_strategy, temperature,
     max_new_tokens, repetition_penalty, top_p

         #_attn_implementation="flash_attention_2",
         trust_remote_code=True).to("cuda")
+@spaces.GPU(duration=180)
 def model_inference(
     image, text, decoding_strategy, temperature,
     max_new_tokens, repetition_penalty, top_p