Spaces:
Paused
Paused
add trust_remote_code=True
Browse files- 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(
|