taming-transformers
taming-transformers copied to clipboard
where is the coco segmentation vqgan pre-train model
when generation segmentation to image i can't find this file "ckpt_path: logs/2020-11-07T00-08-54_cocostuffthings_vqvae_segmentation_bce/checkpoints/last.ckpt"
It's included in the transformer checkpoint. If you want to extract it, run
python scripts/extract_submodel.py logs/2021-01-20T16-04-20_coco_transformer/checkpoints/last.ckpt coco_cond_stage.ckpt cond_stage_model
When trying to train the transformer I found the following error:
ModuleNotFoundError: No module named 'cutlit'
If I change "culit" for "main":
TypeError: cannot unpack non-iterable NoneType object