How to generate different aspect ratio images?

[lyp741]

Great work! I've tried the generation using depth control. But it always generates image with 768*768. Is there any means to generate images with the same as the depth image?

[wuyi2020]

Sorry, our current model only supports generating images with the resolution of 768x768.

[lyp741]

Sorry, our current model only supports generating images with the resolution of 768x768.

Well, actually I've succeeded generating images with arbitrary resolutions by adding tokens to the end of the prompt to force the model to generate arbitrary images, the sample code are as follows:

        h = img_processed.size[1]
        w = img_processed.size[0]
        added_tokens = [
            self.item_processor.token2id(self.item_processor.image_start_token),
            self.item_processor.token2id(self.item_processor.get_n_grids_token(h // 32)),
            self.item_processor.token2id(self.item_processor.get_n_grids_token(w // 32)),
        ]
        prompt = prompt + added_tokens