change Sampler hook

.gitattributes

@@ -1,2 +0,0 @@
-*.png filter=lfs diff=lfs merge=lfs -text
-*.jpg filter=lfs diff=lfs merge=lfs -text

CHANGELOG.md

@@ -1,3 +1,6 @@
+### v2.0.0 - 2024 Feb.29
+- Removed **LFS**
+
 ### v1.5.1 - 2023 Dec.03
 - Bug Fix by. **catboxanon**
 

scripts/cc_callback.py

@@ -1,9 +1,9 @@
-from modules.sd_samplers_kdiffusion import KDiffusionSampler
+from modules.sd_samplers_common import Sampler
 from modules import script_callbacks
 from scripts.cc_scaling import apply_scaling
 from scripts.cc_noise import *
 
-original_callback = KDiffusionSampler.callback_state
+original_callback = Sampler.callback_state
 
 def cc_callback(self, d):
     if not self.vec_cc['enable']:
@@ -60,9 +60,9 @@ def cc_callback(self, d):
 
     return original_callback(self, d)
 
-KDiffusionSampler.callback_state = cc_callback
+Sampler.callback_state = cc_callback
 
 def restore_callback():
-    KDiffusionSampler.callback_state = original_callback
+    Sampler.callback_state = original_callback
 
 script_callbacks.on_script_unloaded(restore_callback)

scripts/cc_version.py

@@ -2,7 +2,7 @@ from modules import script_callbacks
 import modules.scripts as scripts
 import json
 
-VERSION = 'v1.5.1'
+VERSION = 'v2.0.0'
 
 def clean_outdated(EXT_NAME:str):
     with open(scripts.basedir() + '/' + 'ui-config.json', 'r', encoding='utf8') as json_file: