"src/diffusers/models/controlnets/controlnet_sparsectrl.py" did not exist on "766aa50f70120ab8c08c221fb95c68661422fef0"
Unverified Commit 4e138bd5 authored by xkszltl's avatar xkszltl Committed by GitHub
Browse files

Add namespace to avoid conflict with ATen version of channel_shuffle(). (#2206)

Fix https://github.com/pytorch/vision/issues/2193.
parent 7a2d0618
......@@ -80,7 +80,7 @@ struct ShuffleNetV2InvertedResidualImpl : torch::nn::Module {
} else
out = torch::cat({branch1->forward(x), branch2->forward(x)}, 1);
out = channel_shuffle(out, 2);
out = ::vision::models::channel_shuffle(out, 2);
return out;
}
};
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment