Spaces:
Build error
Build error
Update XPart/partgen/partformer_pipeline.py
Browse files
XPart/partgen/partformer_pipeline.py
CHANGED
|
@@ -256,7 +256,7 @@ class PartFormerPipeline(TokenAllocMixin):
|
|
| 256 |
conditioner = vae
|
| 257 |
scheduler = instantiate_from_config(config["scheduler"])
|
| 258 |
config["bbox_predictor"]["params"]["ckpt_path"] = os.path.join(
|
| 259 |
-
ckpt_path, "p3sam.
|
| 260 |
)
|
| 261 |
bbox_predictor = instantiate_from_config(config.get("bbox_predictor", None))
|
| 262 |
model_kwargs = dict(
|
|
|
|
| 256 |
conditioner = vae
|
| 257 |
scheduler = instantiate_from_config(config["scheduler"])
|
| 258 |
config["bbox_predictor"]["params"]["ckpt_path"] = os.path.join(
|
| 259 |
+
ckpt_path, "p3sam.pt"
|
| 260 |
)
|
| 261 |
bbox_predictor = instantiate_from_config(config.get("bbox_predictor", None))
|
| 262 |
model_kwargs = dict(
|