Sunderbraze
/ ComfyBootlegOffload.py
Last active
March 5, 2025 02:03
— forked from city96/ComfyBootlegOffload.py
Add node to offload model as well
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # Force model to always use specified device | |
| # Place in `ComfyUI\custom_nodes` to use | |
| # City96 [Apache2] | |
| # | |
| import types | |
| import torch | |
| import comfy.model_management | |
| class OverrideDevice: | |
| @classmethod |