Leyo commited on
Commit
da9b25b
1 Parent(s): 07bc261

Update app_dialogue.py

Browse files
Files changed (1) hide show
  1. app_dialogue.py +2 -2
app_dialogue.py CHANGED
@@ -27,12 +27,12 @@ MODELS = {
27
  token=os.environ["HF_AUTH_TOKEN"],
28
  revision="73b35c12d88ccc2d9904d9f2f6a755f4973c228c",
29
  ).to(DEVICE),
30
- "288_ter - mix 8 - opt 11'000": AutoModelForCausalLM.from_pretrained(
31
  "HuggingFaceM4/idefics2",
32
  trust_remote_code=True,
33
  torch_dtype=torch.bfloat16,
34
  token=os.environ["HF_AUTH_TOKEN"],
35
- revision="7eccbf5178f85eee8fab9995f31ab12441ce767a",
36
  ).to(DEVICE),
37
  # "285 - continued pretraining on text sft - opt 2'000": AutoModelForCausalLM.from_pretrained(
38
  # "HuggingFaceM4/idefics2",
 
27
  token=os.environ["HF_AUTH_TOKEN"],
28
  revision="73b35c12d88ccc2d9904d9f2f6a755f4973c228c",
29
  ).to(DEVICE),
30
+ "tr_288_cinco_final_sft_sphinx - opt 11'000": AutoModelForCausalLM.from_pretrained(
31
  "HuggingFaceM4/idefics2",
32
  trust_remote_code=True,
33
  torch_dtype=torch.bfloat16,
34
  token=os.environ["HF_AUTH_TOKEN"],
35
+ revision="316ea4acf714760882ad89e364ae1f8c447ae82e",
36
  ).to(DEVICE),
37
  # "285 - continued pretraining on text sft - opt 2'000": AutoModelForCausalLM.from_pretrained(
38
  # "HuggingFaceM4/idefics2",