JinnGame commited on
Commit
10cced1
·
verified ·
1 Parent(s): 9d2d27f

Upload 2 files

Browse files
fp16_Model_Keys_Delete.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import re
2
+ from safetensors.torch import load_file, save_file
3
+
4
+ # 加载 .safetensors 文件
5
+ #file_path = "iniverseMixXLSFWNSFW_flux1DNsfwFp8V10.safetensors"
6
+ file_path = "iniverseMixXLSFWNSFW_flux1DNsfwFp16V12.safetensors"
7
+ model = load_file(file_path)
8
+
9
+ # 定义通配符模式,使用正则表达式匹配要删除的键
10
+ # 例如,删除以 "layer" 开头的所有权重
11
+ pattern = re.compile(r'^text_encoders.*')
12
+ #pattern = re.compile(r'^vae.*')
13
+
14
+ # 找到需要删除的键
15
+ keys_to_delete = [key for key in model.keys() if pattern.match(key)]
16
+
17
+ # 删除匹配的键
18
+ for key in keys_to_delete:
19
+ del model[key]
20
+
21
+ # 保存修改后的模型
22
+ #save_file(model, "iniverseMixXLSFWNSFW_flux1DNsfwFp8_12GV10.safetensors")
23
+ save_file(model, "iniverseMixXLSFWNSFW_flux1DNsfwFp16_22GV12.safetensors")
24
+
25
+ print(f"Deleted {len(keys_to_delete)} keys matching pattern.")
fp8_Model_Keys_Delete.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import re
2
+ from safetensors.torch import load_file, save_file
3
+
4
+ # 加载 .safetensors 文件
5
+ file_path = "iniverseMixXLSFWNSFW_flux1DNsfwFp8V10.safetensors"
6
+ #file_path = "iniverseMixXLSFWNSFW_flux1DNsfwFp16V12.safetensors"
7
+ model = load_file(file_path)
8
+
9
+ # 定义通配符模式,使用正则表达式匹配要删除的键
10
+ # 例如,删除以 "layer" 开头的所有权重
11
+ pattern = re.compile(r'^text_encoders.*')
12
+ #pattern = re.compile(r'^vae.*')
13
+
14
+ # 找到需要删除的键
15
+ keys_to_delete = [key for key in model.keys() if pattern.match(key)]
16
+
17
+ # 删除匹配的键
18
+ for key in keys_to_delete:
19
+ del model[key]
20
+
21
+ # 保存修改后的模型
22
+ save_file(model, "iniverseMixXLSFWNSFW_flux1DNsfwFp8_12GV10.safetensors")
23
+ #save_file(model, "iniverseMixXLSFWNSFW_flux1DNsfwFp16_22GV12.safetensors")
24
+
25
+ print(f"Deleted {len(keys_to_delete)} keys matching pattern.")