dwb2023 commited on
Commit
cac741c
·
verified ·
1 Parent(s): 055b476

add trust_remote_code=True

Browse files
Files changed (1) hide show
  1. app_dialogue.py +1 -0
app_dialogue.py CHANGED
@@ -32,6 +32,7 @@ MODELS = {
32
  "Salesforce/xgen-mm-phi3-mini-instruct-interleave-r-v1.5",
33
  torch_dtype=torch.bfloat16,
34
  _attn_implementation="flash_attention_2",
 
35
  ).to(DEVICE),
36
  }
37
  PROCESSOR = AutoProcessor.from_pretrained(
 
32
  "Salesforce/xgen-mm-phi3-mini-instruct-interleave-r-v1.5",
33
  torch_dtype=torch.bfloat16,
34
  _attn_implementation="flash_attention_2",
35
+ trust_remote_code=True
36
  ).to(DEVICE),
37
  }
38
  PROCESSOR = AutoProcessor.from_pretrained(