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

19f3bfa

·

verified ·

1 Parent(s): ad720cd

Update README.md

Files changed (1) hide show

README.md +0 -1

README.md CHANGED Viewed

@@ -44,7 +44,6 @@ from transformers import AutoModelForCausalLM
 model = AutoModelForCausalLM.from_pretrained(
     "tiiuae/falcon-perception",
     trust_remote_code=True,
-    torch_dtype=torch.bfloat16,
     device_map={"": "cuda:0"},
 )

 model = AutoModelForCausalLM.from_pretrained(
     "tiiuae/falcon-perception",
     trust_remote_code=True,
     device_map={"": "cuda:0"},
 )