mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-02-25 17:23:58 +00:00
Update controlnet.py
This commit is contained in:
@@ -40,6 +40,7 @@ class ControlBase:
|
||||
self.timestep_percent_range = (0.0, 1.0)
|
||||
self.global_average_pooling = False
|
||||
self.timestep_range = None
|
||||
self.transformer_options = {}
|
||||
|
||||
if device is None:
|
||||
device = ldm_patched.modules.model_management.get_torch_device()
|
||||
|
||||
Reference in New Issue
Block a user