jiauzhang / gigagan Goto Github PK

rahulbhalley@192 GigaGAN % python3 generate.py
Traceback (most recent call last):
  File "/Users/rahulbhalley/Desktop/Hide-and-Seek/image-synth/GigaGAN/generate.py", line 17, in <module>
    images = g(z, text_embeds)[0]
             ^^^^^^^^^^^^^^^^^
  File "/opt/homebrew/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/rahulbhalley/Desktop/Hide-and-Seek/image-synth/GigaGAN/model.py", line 140, in forward
    styles = [self.style(s) for s in styles]
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/rahulbhalley/Desktop/Hide-and-Seek/image-synth/GigaGAN/model.py", line 140, in <listcomp>
    styles = [self.style(s) for s in styles]
              ^^^^^^^^^^^^^
  File "/opt/homebrew/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/homebrew/lib/python3.11/site-packages/torch/nn/modules/container.py", line 217, in forward
    input = module(input)
            ^^^^^^^^^^^^^
  File "/opt/homebrew/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/rahulbhalley/Desktop/Hide-and-Seek/image-synth/GigaGAN/layers.py", line 11, in forward
    return input * torch.rsqrt(torch.mean(input ** 2, dim=1, keepdim=True) + 1e-8)
                               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)