allenai
/

MolmoAct-7B-D-LIBERO-Spatial-0812

image-text-to-text

Model card Files Files and versions

hqfang commited on Aug 31

Commit

cca2edd

·

verified ·

1 Parent(s): 2e8eebd

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-Spatial-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",
 )