删除免费丹里面他们自己添加的键值键名

day270010678

星级打分

• 1
• 2
• 3
• 4
• 5

平均分:NAN  参与人数:0  我的评分:未评

大家公布的免费丹，很多都添加了一些自定义的东西，这是可以理解的，但是里面的语言过激，属实有点不文明。比方说这个里面的



==---------------- Model Options -----------------==

==        gan_patch_size: 36                      ==
==              gan_dims: 16                      ==
==              猫之汉化免费丹 : 564646676               ==
==                  盗卖死妈: 生孩子全身都是痔疮               ==
==                  请自重！: 请自重！                    ==
==                  AI王子: dapixiao                ==
==                  西行之巅: 奥贝赛维                    ==
==                都是死妈骗子: 小心！                     ==


里面的这些属实有点不文明

==                  盗卖死妈: 生孩子全身都是痔疮               ==
==                  请自重！: 请自重！                    ==
==                  AI王子: dapixiao                ==
==                  西行之巅: 奥贝赛维                    ==
==                都是死妈骗子: 小心！                     ==


行了直接说方法：
models里面的都是pickle生成的序列化的数据，知道这个加载修改就行了。
我给大家提供一种脚本，代码如下面，复制后存为py文件就行了。

1. import pickle
   
2. from pathlib import Path
   
3. import sys
   
4. 
   
5. 
   
6. def load_model(model_path):
   
7.     """加载模型文件"""
   
8.     try:
   
9.         model_data = Path(model_path).read_bytes()
   
10.         model = pickle.loads(model_data)
    
11.         return model
    
12.     except FileNotFoundError:
    
13.         print(f"错误: 找不到模型文件 {model_path}")
    
14.         return None
    
15.     except Exception as e:
    
16.         print(f"加载模型时出错: {e}")
    
17.         return None
    
18. 
    
19. 
    
20. def save_model(model, model_path):
    
21.     """保存模型到文件"""
    
22.     try:
    
23.         model_data = pickle.dumps(model)
    
24.         Path(model_path).write_bytes(model_data)
    
25.         print(f"模型已成功保存到 {model_path}")
    
26.         return True
    
27.     except Exception as e:
    
28.         print(f"保存模型时出错: {e}")
    
29.         return False
    
30. 
    
31. 
    
32. def remove_option_from_model(model_path, option_key):
    
33.     """从模型中删除指定的options配置并保存"""
    
34.     # 加载模型
    
35.     model = load_model(model_path)
    
36.     if model is None:
    
37.         return False
    
38. 
    
39.     # 检查是否存在options
    
40.     if 'options' in model:
    
41.         options = model['options']
    
42.         if option_key in options:
    
43.             # 删除指定的option
    
44.             removed_value = options.pop(option_key)
    
45.             print(f"已从模型中删除option '{option_key}': {removed_value}")
    
46. 
    
47.             # 保存修改后的模型
    
48.             return save_model(model, model_path)
    
49.         else:
    
50.             print(f"模型中未找到option '{option_key}'")
    
51.             return False
    
52.     else:
    
53.         print("模型中未找到options配置")
    
54.         return False
    
55. 
    
56. 
    
57. def list_options(model_path):
    
58.     """列出模型中的所有options"""
    
59.     model = load_model(model_path)
    
60.     if model is None:
    
61.         return
    
62. 
    
63.     if 'options' in model:
    
64.         print("当前模型中的options:")
    
65.         for key, value in model['options'].items():
    
66.             print(f"  {key}: {value}")
    
67.     else:
    
68.         print("模型中未找到options配置")
    
69. 
    
70. 
    
71. def main():
    
72.     """主函数"""
    
73.     print("=== 模型Options管理工具 ===")
    
74. 
    
75.     # 获取模型文件路径
    
76.     model_path = input("请输入模型文件路径 (默认: XSeg_data.dat): ").strip()
    
77.     if not model_path:
    
78.         model_path = r"\DeepFaceLab_NVIDIA_RTX3000_series\workspace\model\XSeg_data.dat"
    
79. 
    
80.     while True:
    
81.         print("\n请选择操作:")
    
82.         print("1. 查看所有options")
    
83.         print("2. 删除指定option")
    
84.         print("3. 退出")
    
85. 
    
86.         choice = input("请输入选择 (1-3): ").strip()
    
87. 
    
88.         if choice == '1':
    
89.             list_options(model_path)
    
90. 
    
91.         elif choice == '2':
    
92.             list_options(model_path)
    
93.             option_key = input("请输入要删除的option键名: ").strip()
    
94.             if option_key:
    
95.                 confirm = input(f"确认要删除option '{option_key}'吗? (y/N): ").strip().lower()
    
96.                 if confirm == 'y' or confirm == 'yes':
    
97.                     remove_option_from_model(model_path, option_key)
    
98.                 else:
    
99.                     print("操作已取消")
    
100.             else:
     
101.                 print("未输入有效的option键名")
     
102. 
     
103.         elif choice == '3':
     
104.             print("再见!")
     
105.             break
     
106. 
     
107.         else:
     
108.             print("无效选择，请重新输入")
     
109. 
     
110. 
     
111. if __name__ == "__main__":
     
112.     main()
     

复制代码

把上面的复制到文本里面 存为xxx.py文件就行了。

然后在cmd里面用deepfacelab里面的python调用这个py文件运行就行了

比方说我这个命令行里面   输入

1. D:\AI\DeepFaceLab_NVIDIA_RTX3000_series\_internal\python-3.6.8\python.exe D:\AI\DeepFaceLab_PyTorch\_internal\修改模型里面的键名.py

复制代码

D:\AI\DeepFaceLab_NVIDIA_RTX3000_series\_internal\python-3.6.8\python.exe 是你的deepfacelab目录里面的python-3.6.8\python.exe
D:\AI\DeepFaceLab_PyTorch\_internal\修改模型里面的键名.py 是上面复制的代码存为的xxx.py的路径


然后按照里面的提示逐步操作就行了