Spaces:
Running
on
Zero
Running
on
Zero
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 |
|