Spaces:
Runtime error
Runtime error
VictorSanh
commited on
Commit
•
c4e5100
1
Parent(s):
33c31cc
Update app_dialogue.py
Browse files- app_dialogue.py +0 -4
app_dialogue.py
CHANGED
@@ -33,12 +33,10 @@ MODELS = {
|
|
33 |
torch_dtype=torch.bfloat16,
|
34 |
_attn_implementation="flash_attention_2",
|
35 |
trust_remote_code=True,
|
36 |
-
token=os.environ["HF_AUTH_TOKEN"],
|
37 |
).to(DEVICE),
|
38 |
}
|
39 |
PROCESSOR = AutoProcessor.from_pretrained(
|
40 |
"HuggingFaceM4/idefics2-8b",
|
41 |
-
token=os.environ["HF_AUTH_TOKEN"],
|
42 |
)
|
43 |
|
44 |
SYSTEM_PROMPT = [
|
@@ -169,8 +167,6 @@ EXAMPLES = [
|
|
169 |
],
|
170 |
]
|
171 |
|
172 |
-
API_TOKEN = os.getenv("HF_AUTH_TOKEN")
|
173 |
-
HF_WRITE_TOKEN = os.getenv("HF_WRITE_TOKEN")
|
174 |
BOT_AVATAR = "IDEFICS_logo.png"
|
175 |
|
176 |
|
|
|
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(
|
39 |
"HuggingFaceM4/idefics2-8b",
|
|
|
40 |
)
|
41 |
|
42 |
SYSTEM_PROMPT = [
|
|
|
167 |
],
|
168 |
]
|
169 |
|
|
|
|
|
170 |
BOT_AVATAR = "IDEFICS_logo.png"
|
171 |
|
172 |
|