warnings

[mycomedico]

Hi there! I really like your colorization model. I'm using it on old images and I think it works better than the other currently available options, except sometimes red color is too strong. When I run it as python script I get a couple of warnings I want to ask if there are any ways to avoid these warnings thanks!

2024-03-15 17:43:21,771 - modelscope - INFO - PyTorch version 2.2.1+rocm5.7 Found.
2024-03-15 17:43:21,797 - modelscope - INFO - Loading ast index from /home/rubing/.cache/modelscope/ast_indexer
2024-03-15 17:43:21,824 - modelscope - INFO - Loading done! Current index file version is 1.13.1, with md5 bf5a0cac5e5265c888d1b6453711fc56 and a total number of 972 components indexed
2024-03-15 17:43:25,009 - modelscope - WARNING - Model revision not specified, use revision: v1.02
2024-03-15 17:43:25,743 - modelscope - INFO - initiate model from /home/rubing/.cache/modelscope/hub/damo/cv_ddcolor_image-colorization
2024-03-15 17:43:25,745 - modelscope - INFO - initiate model from location /home/rubing/.cache/modelscope/hub/damo/cv_ddcolor_image-colorization.
2024-03-15 17:43:25,748 - modelscope - INFO - initialize model from /home/rubing/.cache/modelscope/hub/damo/cv_ddcolor_image-colorization
2024-03-15 17:43:29,373 - modelscope - INFO - Loading DDColor model from /home/rubing/.cache/modelscope/hub/damo/cv_ddcolor_image-colorization/pytorch_model.pt, with param key: [params].
2024-03-15 17:43:29,658 - modelscope - INFO - load model done.
2024-03-15 17:43:29,737 - modelscope - WARNING - No preprocessor field found in cfg.
2024-03-15 17:43:29,779 - modelscope - WARNING - No val key and type key found in preprocessor domain of configuration.json file.
2024-03-15 17:43:29,781 - modelscope - WARNING - Cannot find available config to build preprocessor at mode inference, current config: {'model_dir': '/home/rubing/.cache/modelscope/hub/damo/cv_ddcolor_image-colorization'}. trying to build by task and model information.
2024-03-15 17:43:29,782 - modelscope - WARNING - No preprocessor key ('ddcolor', 'image-colorization') found in PREPROCESSOR_MAP, skip building preprocessor.
2024-03-15 17:43:29,785 - modelscope - INFO - load model done

[piddnad]

Hello, thanks for liking our work!

To avoid these warnings, you can run the model through inference locally or inference with HuggingFace.

Also, regarding the color preference of coloring, try the different models here to see if they work better in specific scenarios.