Spaces:
Runtime error
Runtime error
Commit
•
d06562a
1
Parent(s):
c569980
path update
Browse files
the_updated_app_with_tfrm_integration.py
CHANGED
@@ -19,7 +19,7 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
|
|
19 |
DEVICE = torch.device("cuda")
|
20 |
MODELS = {
|
21 |
"idefics2": Idefics2ForConditionalGeneration.from_pretrained(
|
22 |
-
"/
|
23 |
torch_dtype=torch.bfloat16,
|
24 |
_attn_implementation="flash_attention_2",
|
25 |
trust_remote_code=True,
|
|
|
19 |
DEVICE = torch.device("cuda")
|
20 |
MODELS = {
|
21 |
"idefics2": Idefics2ForConditionalGeneration.from_pretrained(
|
22 |
+
"HuggingFaceM4/idefics2-tfrm-compatible",
|
23 |
torch_dtype=torch.bfloat16,
|
24 |
_attn_implementation="flash_attention_2",
|
25 |
trust_remote_code=True,
|