allenai
/

MolmoAct-7B-D-LIBERO-Object-0812

image-text-to-text

Model card Files Files and versions

hqfang commited on Aug 31

Commit

235264c

·

verified ·

1 Parent(s): dc0553f

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -62,7 +62,7 @@ ckpt = "allenai/MolmoAct-7B-D-LIBERO-Object-0812"
 processor = AutoProcessor.from_pretrained(
     ckpt,
     trust_remote_code=True,
-    torch_dtype="auto",
     device_map="auto",
     padding_side="left",
 )
@@ -71,7 +71,7 @@ processor = AutoProcessor.from_pretrained(
 model = AutoModelForImageTextToText.from_pretrained(
     ckpt,
     trust_remote_code=True,
-    torch_dtype="auto",
     device_map="auto",
 )

 processor = AutoProcessor.from_pretrained(
     ckpt,
     trust_remote_code=True,
+    torch_dtype="bfloat16",
     device_map="auto",
     padding_side="left",
 )
 model = AutoModelForImageTextToText.from_pretrained(
     ckpt,
     trust_remote_code=True,
+    torch_dtype="bfloat16",
     device_map="auto",
 )