Skip to content

Commit

Permalink
onediff control lora remove device attr (#1155)
Browse files Browse the repository at this point in the history
<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->

## Summary by CodeRabbit

- **New Features**
- Enhanced configuration process for the control model, improving setup
and error handling.
	- Streamlined management of model attributes and device settings.

- **Bug Fixes**
- Improved error handling during weight assignment in the diffusion
model.

- **Refactor**
	- Updated method signatures for better clarity and maintainability.

<!-- end of auto-generated comment: release notes by coderabbit.ai -->
  • Loading branch information
marigoold authored Dec 3, 2024
1 parent 1bf803d commit 4dde754
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions onediff_comfy_nodes/modules/oneflow/onediff_controlnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,6 @@ def from_controllora(
c = cls(
controlnet.control_weights,
global_average_pooling=controlnet.global_average_pooling,
device=controlnet.device,
)
controlnet.copy_to(c)
c._oneflow_model = None
Expand Down Expand Up @@ -117,7 +116,6 @@ def copy(self):
c = OneDiffControlLora(
self.control_weights,
global_average_pooling=self.global_average_pooling,
device=self.device,
)
self.copy_to(c)
c._oneflow_model = self._oneflow_model
Expand Down

0 comments on commit 4dde754

Please sign in to comment.