search

willisma / SiT

Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"
https://scalable-interpolant.github.io/
MIT License
662 stars 35 forks source link

AttributeError: 'Namespace' object has no attribute 'model' #4

Closed stefenmax closed 7 months ago

stefenmax commented 8 months ago

Hi, when I want to use python sample.py ODE --image-size 256 --seed 1 to sample This error raised Traceback (most recent call last): File "/mnt/new_ssd/stefen/code/SiT/sample.py", line 146, in main(mode, args) File "/mnt/new_ssd/stefen/code/SiT/sample.py", line 28, in main assert args.model == "SiT-XL/2", "Only SiT-XL/2 models are available for auto-download." AttributeError: 'Namespace' object has no attribute 'model'

willisma commented 7 months ago

Hi, This is due to a small typo in sample.py. It should be fixed by now.