Skip to content

Comments

Fix kohya-ss OFT network wrong device for eye and constraint#14726

Merged
AUTOMATIC1111 merged 1 commit intoAUTOMATIC1111:devfrom
v0xie:fix-oft-device
Jan 23, 2024
Merged

Fix kohya-ss OFT network wrong device for eye and constraint#14726
AUTOMATIC1111 merged 1 commit intoAUTOMATIC1111:devfrom
v0xie:fix-oft-device

Conversation

@v0xie
Copy link
Contributor

@v0xie v0xie commented Jan 22, 2024

Description

This PR fixes a "Expected tensors to be on same device" error for kohya-ss OFT networks.
We send eye and self.constraint to the same device as oft_blocks before performing operations with them.

Screenshots/videos:

N/A

Checklist:

@v0xie v0xie requested a review from AUTOMATIC1111 as a code owner January 22, 2024 11:00
@AUTOMATIC1111 AUTOMATIC1111 merged commit 358e9e2 into AUTOMATIC1111:dev Jan 23, 2024
@w-e-w w-e-w mentioned this pull request Feb 17, 2024
@pawel665j pawel665j mentioned this pull request Apr 16, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants