murcherful commited on
Commit
8231326
·
verified ·
1 Parent(s): a9923b4

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.ckpt"
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(