785bb48
1
2
3
4
CUDA_VISIBLE_DEVICES=0 \ python3 inference/colorization_pipline.py \ --input ./assets/test_images --output ./colorize_output \ --model_path modelscope/damo/cv_ddcolor_image-colorization/pytorch_model.pt
