search

vislearn / ControlNet-XS

Apache License 2.0
411 stars 12 forks source link

Cannnot use sd21 model #10

Closed SoftologyPro closed 9 months ago

SoftologyPro commented 9 months ago

First when trying the use sd21 I got

cannot import name 'rank_zero_only' from 'pytorch_lightning.utilities.distributed'

which needed pytorch-lightning to be rolled back to v1.6.5 to get past that.

But now I get...

Traceback (most recent call last):
  File "D:\ControlNet-XS\ControlNet-XS.py", line 69, in <module>
    samples, controls = cu.get_sdxl_sample(
  File "D:\venv\lib\site-packages\torch\utils\_contextlib.py", line 115, in decorate_context
    return func(*args, **kwargs)
  File "D:ControlNet-XS\scripts\control_utils.py", line 180, in get_sdxl_sample
    model.sampler.num_steps = ddim_steps
  File "D:\venv\lib\site-packages\torch\nn\modules\module.py", line 1614, in __getattr__
    raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'TwoStreamControlLDM' object has no attribute 'sampler'. Did you mean: 'sample'?

Any ideas on what needs to be tweaked for sd21 to work? sdxl works fine. Thanks.