danielroich / PTI

Official Implementation for "Pivotal Tuning for Latent-based editing of Real Images" (ACM TOG 2022) https://arxiv.org/abs/2106.05744

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Failed to run 512x512 images

alexey-gorkor opened this issue · comments

In w_projector.py we have a hardcoded size that caused me an error. I encountered this issue when I used 512x512 images.

I propose to set it as w_opt.repeat([1, G.mapping.num_ws, 1])