Remy commited on
Commit
e4acd8c
·
verified ·
1 Parent(s): 81f728a

Update ActionMesh space

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -296,6 +296,8 @@ def load_pipeline_cpu() -> GradioPipeline:
296
  pipeline = GradioPipeline(
297
  config_name="actionmesh.yaml",
298
  config_dir=config_dir,
 
 
299
  )
300
  print("Pipeline loaded on CPU successfully.")
301
  return pipeline
 
296
  pipeline = GradioPipeline(
297
  config_name="actionmesh.yaml",
298
  config_dir=config_dir,
299
+ lazy_loading=False,
300
+ dtype=torch.bfloat16,
301
  )
302
  print("Pipeline loaded on CPU successfully.")
303
  return pipeline