Zaherrr commited on
Commit
563b496
·
verified ·
1 Parent(s): 636931b

removed the @spaces.GPU

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ model = Idefics3ForConditionalGeneration.from_pretrained("HuggingFaceM4/Idefics3
20
  BAD_WORDS_IDS = processor.tokenizer(["<image>", "<fake_token_around_image>"], add_special_tokens=False).input_ids
21
  EOS_WORDS_IDS = [processor.tokenizer.eos_token_id]
22
 
23
- @spaces.GPU
24
  def model_inference(
25
  images, text, assistant_prefix, decoding_strategy, temperature, max_new_tokens,
26
  repetition_penalty, top_p
 
20
  BAD_WORDS_IDS = processor.tokenizer(["<image>", "<fake_token_around_image>"], add_special_tokens=False).input_ids
21
  EOS_WORDS_IDS = [processor.tokenizer.eos_token_id]
22
 
23
+ # @spaces.GPU
24
  def model_inference(
25
  images, text, assistant_prefix, decoding_strategy, temperature, max_new_tokens,
26
  repetition_penalty, top_p