Spaces:
Running on Zero
Running on Zero
Update ActionMesh space
Browse files
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
|