Hugging Face's logo

Spaces:
HuggingFaceM4
/
idefics2_playground
Running on Zero

App Files Community
11
VictorSanh commited on
Commit
2e44cff
·
verified ·
1 Parent(s): 598bc19

Update app_dialogue.py

Browse files
Files changed (1) hide show
  1. app_dialogue.py +2 -2
app_dialogue.py CHANGED
@@ -26,12 +26,12 @@ MODELS = {
26
  token=os.environ["HF_AUTH_TOKEN"],
27
  revision="a1bc6a2b0f74cde25844144f602dde2808a564d9",
28
  ).to(DEVICE),
29
- "286 - mix6 tables - opt 20'000": AutoModelForCausalLM.from_pretrained(
30
  "HuggingFaceM4/idefics2",
31
  trust_remote_code=True,
32
  torch_dtype=torch.bfloat16,
33
  token=os.environ["HF_AUTH_TOKEN"],
34
- revision="b473d49caa964991b40b79fe7cb27d51d4d023f6",
35
  ).to(DEVICE),
36
  # "285 - continued pretraining on text sft - opt 2'000": AutoModelForCausalLM.from_pretrained(
37
  # "HuggingFaceM4/idefics2",
 
26
  token=os.environ["HF_AUTH_TOKEN"],
27
  revision="a1bc6a2b0f74cde25844144f602dde2808a564d9",
28
  ).to(DEVICE),
29
+ "282 - mix 5 - opt 18'500": AutoModelForCausalLM.from_pretrained(
30
  "HuggingFaceM4/idefics2",
31
  trust_remote_code=True,
32
  torch_dtype=torch.bfloat16,
33
  token=os.environ["HF_AUTH_TOKEN"],
34
+ revision="28296a42b9b31bf6a10322ba3e35bb0170f0b7f6",
35
  ).to(DEVICE),
36
  # "285 - continued pretraining on text sft - opt 2'000": AutoModelForCausalLM.from_pretrained(
37
  # "HuggingFaceM4/idefics2",