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,
|