Hugging Face's logo

Spaces:
HuggingFaceM4
/
idefics2_playground
Running on Zero

App Files Community
11
Leyo commited on
Commit
016ec10
•
1 Parent(s): 01ea54d

update

Browse files
Files changed (1) hide show
  1. app_dialogue.py +10 -10
app_dialogue.py CHANGED
@@ -35,22 +35,22 @@ 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
- # revision="bb460e58294bcb02430df9fd126b3c522f867d83"
45
- # ).to(DEVICE),
46
- "idefics2-8b-chatty (chat-50)": Idefics2ForConditionalGeneration.from_pretrained(
47
  "HuggingFaceM4/idefics2-8b-chatty",
48
  torch_dtype=torch.bfloat16,
49
  _attn_implementation="flash_attention_2",
50
  trust_remote_code=True,
51
  token=os.environ["HF_AUTH_TOKEN"],
52
- revision="1d57ffb705199370f7875667cc8f54abd09b2004"
53
  ).to(DEVICE),
 
 
 
 
 
 
 
 
54
  }
55
  PROCESSOR = AutoProcessor.from_pretrained(
56
  "HuggingFaceM4/idefics2-8b",
 
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
+ revision="bb460e58294bcb02430df9fd126b3c522f867d83"
45
  ).to(DEVICE),
46
+ # "idefics2-8b-chatty (chat-50)": Idefics2ForConditionalGeneration.from_pretrained(
47
+ # "HuggingFaceM4/idefics2-8b-chatty",
48
+ # torch_dtype=torch.bfloat16,
49
+ # _attn_implementation="flash_attention_2",
50
+ # trust_remote_code=True,
51
+ # token=os.environ["HF_AUTH_TOKEN"],
52
+ # revision="1d57ffb705199370f7875667cc8f54abd09b2004"
53
+ # ).to(DEVICE),
54
  }
55
  PROCESSOR = AutoProcessor.from_pretrained(
56
  "HuggingFaceM4/idefics2-8b",