diff --git a/extensions-builtin/LDSR/ldsr_model_arch.py b/extensions-builtin/LDSR/ldsr_model_arch.py index 7cac36ce5..e7400a34c 100644 --- a/extensions-builtin/LDSR/ldsr_model_arch.py +++ b/extensions-builtin/LDSR/ldsr_model_arch.py @@ -104,6 +104,7 @@ class LDSR: return logs def super_resolution(self, image, steps=100, target_scale=2, half_attention=False): + previous_clip = sd_hijack.model_hijack.clip model = self.load_model_from_config(half_attention) # Run settings @@ -151,6 +152,8 @@ class LDSR: gc.collect() devices.torch_gc() + sd_hijack.model_hijack.clip = previous_clip + return a