search

vislearn / ControlNet-XS

Apache License 2.0
410 stars 12 forks source link

Where ControlNet-XS model code file in whole code ? #3

Closed motion-dream closed 9 months ago

motion-dream commented 9 months ago

Your work is good! I want to learn from your work. Can you tell ControlNet-XS model position, I can't find it. Thank you. @_@

Sipirius commented 9 months ago

You can see in the inference config file what module is loaded. For control of SDXL it is 

network_config:
      target: sgm.modules.diffusionmodules.twoStreamControl.TwoStreamControlNet

for control of SDv2.1 it is 

control_stage_config:
      target: ldm.modules.diffusionmodules.twoStreamControl.TwoStreamControlNet

cheers