robotics-diffusion-transformer
/

rdt-1b

Inference Endpoints

Model card Files Files and versions Community

robotics-diffusion-transformer commited on Sep 10

Commit

46a7f5b

•

1 Parent(s): dabf2e4

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -3,7 +3,7 @@ license: mit
 language:
 - en
 pipeline_tag: robotics
-library_name: diffusers
 tags:
 - robotics
 - pytorch
@@ -66,6 +66,7 @@ model = create_model(
     args=config,
     dtype=torch.bfloat16,
     pretrained_vision_encoder_name_or_path=pretrained_vision_encoder_name_or_path,
     control_frequency=25,
 )
 # Start inference process

 language:
 - en
 pipeline_tag: robotics
+library_name: transformers
 tags:
 - robotics
 - pytorch
     args=config,
     dtype=torch.bfloat16,
     pretrained_vision_encoder_name_or_path=pretrained_vision_encoder_name_or_path,
+    pretrained='robotics-diffusion-transformer/rdt-1b',
     control_frequency=25,
 )
 # Start inference process