Sergidev commited on
Commit
fb27ec9
·
1 Parent(s): f2e7b3a
txt2panoimg/text_to_360panorama_image_pipeline.py CHANGED
@@ -207,4 +207,7 @@ class Text2360PanoramaImagePipeline(DiffusionPipeline):
207
  ).images[0]
208
  print('finished')
209
 
 
 
 
210
  return output_img
 
207
  ).images[0]
208
  print('finished')
209
 
210
+ # Ensure output is in PNG format
211
+ if isinstance(output_img, Image.Image):
212
+ output_img = output_img.convert('RGB')
213
  return output_img