bruefire

Hi @Revanthraja. Please use the inference.py script. ``` python inference.py --model '/path/to/checkpoint/' --prompt 'apple' -n "orange" ``` The usage are written in the readme file.

You should have executed a command like the following: ``` python train.py --config configs/my_train_config.yaml ``` The "output_dir" in config.yaml used by your training, is where checkpoints are saved. Please check...

Is there a 'scheduler' directory in the 'train_2023-07-13T11-37-44' ? That directory must contain 'scheduler_config.json', but it probably disappeared for any reason.