Segmentation fault error when doing panel-to-image

ali-vilab / Ranni

https://ranni-t2i.github.io/Ranni/

Apache License 2.0

207 stars 15 forks source link

Segmentation fault error when doing panel-to-image #5

Open rivaldophilip opened 5 months ago

rivaldophilip commented 5 months ago

Segmentation fault error when doing panel-to-image

Error also happens on sample images / generation

rivaldophilip commented 5 months ago

More details using faulthandler

thss15fyt commented 5 months ago

It seems the fault happens at the first time of NN computing in stage 2. Could you try to feed forward the panel conditioner in ranni/panel_conditioner.py with input control = (torch.rand(1, 2, 3, 96, 96).cuda(), torch.rand(1, 2, 1024).cuda())?