Hugging Face's logo

Spaces:
HuggingFaceM4
/
idefics2_playground
Runtime error

App Files Community
11
Leyo commited on
Commit
87af16b
·
1 Parent(s): ab6fa4a

Revert "Update app_dialogue.py"

Browse files

This reverts commit ab6fa4ace67c17972eb65e147e94bd1aca13cb97.

Files changed (1) hide show
  1. app_dialogue.py +0 -7
app_dialogue.py CHANGED
@@ -35,13 +35,6 @@ MODELS = {
35
  # trust_remote_code=True,
36
  # token=os.environ["HF_AUTH_TOKEN"],
37
  # ).to(DEVICE),
38
- "idefics2-8b-chatty (chat-600)": Idefics2ForConditionalGeneration.from_pretrained(
39
- "HuggingFaceM4/idefics2-8b-chatty",
40
- torch_dtype=torch.bfloat16,
41
- _attn_implementation="flash_attention_2",
42
- trust_remote_code=True,
43
- token=os.environ["HF_AUTH_TOKEN"],
44
- ).to(DEVICE),
45
  "idefics2-8b-chatty (chat-50)": Idefics2ForConditionalGeneration.from_pretrained(
46
  "HuggingFaceM4/idefics2-8b-chatty",
47
  torch_dtype=torch.bfloat16,
 
35
  # trust_remote_code=True,
36
  # token=os.environ["HF_AUTH_TOKEN"],
37
  # ).to(DEVICE),
 
 
 
 
 
 
 
38
  "idefics2-8b-chatty (chat-50)": Idefics2ForConditionalGeneration.from_pretrained(
39
  "HuggingFaceM4/idefics2-8b-chatty",
40
  torch_dtype=torch.bfloat16,