[revised] change some dtype behaviors based on community feedbacks

only influence old devices like 1080/70/60/50.
please remove cmd flags if you are on 1080/70/60/50 and previously used many cmd flags to tune performance
This commit is contained in:
layerdiffusion
2024-08-21 10:23:38 -07:00
parent 1419ef29aa
commit 4e3c78178a
4 changed files with 37 additions and 92 deletions
+1 -1
View File
@@ -438,7 +438,7 @@ class ControlLora(ControlNet):
self.manual_cast_dtype = model.computation_dtype
with using_forge_operations(operations=ControlLoraOps, dtype=dtype):
with using_forge_operations(operations=ControlLoraOps, dtype=dtype, manual_cast_enabled=self.manual_cast_dtype != dtype):
self.control_model = cldm.ControlNet(**controlnet_config)
self.control_model.to(device=memory_management.get_torch_device(), dtype=dtype)