tiiuae
/

Falcon-Perception

Mask Generation

falcon_perception

text-generation

vision-language

open-vocabulary

Model card Files Files and versions

yasserDahou commited on Mar 29

Commit

9b053e0

·

verified ·

1 Parent(s): d8138fe

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -45,7 +45,7 @@ model = AutoModelForCausalLM.from_pretrained(
     "tiiuae/falcon-perception",
     trust_remote_code=True,
     torch_dtype=torch.bfloat16,
-    device_map="auto",
 )
 image = Image.open("photo.jpg")

     "tiiuae/falcon-perception",
     trust_remote_code=True,
     torch_dtype=torch.bfloat16,
+    device_map={"": "cuda:0"},
 )
 image = Image.open("photo.jpg")