-
Notifications
You must be signed in to change notification settings - Fork 25
/
clonerepo_experimental.py
222 lines (186 loc) · 9.89 KB
/
clonerepo_experimental.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
import os
import subprocess
import shutil
from concurrent.futures import ThreadPoolExecutor, as_completed
from tqdm.notebook import tqdm
from pathlib import Path
import requests
def run_script():
def run_cmd(cmd):
process = subprocess.run(cmd, shell=True, check=True, text=True)
return process.stdout
# Change the current directory to /content/
os.chdir('/content/')
print("Changing dir to /content/")
# Your function to edit the file
def edit_file(file_path):
temp_file_path = "/tmp/temp_file.py"
changes_made = False
with open(file_path, "r") as file, open(temp_file_path, "w") as temp_file:
previous_line = ""
second_previous_line = ""
for line in file:
new_line = line.replace("value=160", "value=128")
if new_line != line:
print("Replaced 'value=160' with 'value=128'")
changes_made = True
line = new_line
new_line = line.replace("crepe hop length: 160", "crepe hop length: 128")
if new_line != line:
print("Replaced 'crepe hop length: 160' with 'crepe hop length: 128'")
changes_made = True
line = new_line
new_line = line.replace("value=0.88", "value=0.75")
if new_line != line:
print("Replaced 'value=0.88' with 'value=0.75'")
changes_made = True
line = new_line
if "label=i18n(\"输入源音量包络替换输出音量包络融合比例，越靠近1越使用输出包络\")" in previous_line and "value=1," in line:
new_line = line.replace("value=1,", "value=0.25,")
if new_line != line:
print("Replaced 'value=1,' with 'value=0.25,' based on the condition")
changes_made = True
line = new_line
if "label=i18n(\"总训练轮数total_epoch\")" in previous_line and "value=20," in line:
new_line = line.replace("value=20,", "value=500,")
if new_line != line:
print("Replaced 'value=20,' with 'value=500,' based on the condition for DEFAULT EPOCH")
changes_made = True
line = new_line
if 'choices=["pm", "harvest", "dio", "crepe", "crepe-tiny", "mangio-crepe", "mangio-crepe-tiny"], # Fork Feature. Add Crepe-Tiny' in previous_line:
if 'value="pm",' in line:
new_line = line.replace('value="pm",', 'value="mangio-crepe",')
if new_line != line:
print("Replaced 'value=\"pm\",' with 'value=\"mangio-crepe\",' based on the condition")
changes_made = True
line = new_line
new_line = line.replace('label=i18n("输入训练文件夹路径"), value="E:\\\\语音音频+标注\\\\米津玄师\\\\src"', 'label=i18n("输入训练文件夹路径"), value="/content/dataset/"')
if new_line != line:
print("Replaced 'label=i18n(\"输入训练文件夹路径\"), value=\"E:\\\\语音音频+标注\\\\米津玄师\\\\src\"' with 'label=i18n(\"输入训练文件夹路径\"), value=\"/content/dataset/\"'")
changes_made = True
line = new_line
if 'label=i18n("是否仅保存最新的ckpt文件以节省硬盘空间"),' in second_previous_line:
if 'value=i18n("否"),' in line:
new_line = line.replace('value=i18n("否"),', 'value=i18n("是"),')
if new_line != line:
print("Replaced 'value=i18n(\"否\"),' with 'value=i18n(\"是\"),' based on the condition for SAVE ONLY LATEST")
changes_made = True
line = new_line
if 'label=i18n("是否在每次保存时间点将最终小模型保存至weights文件夹"),' in second_previous_line:
if 'value=i18n("否"),' in line:
new_line = line.replace('value=i18n("否"),', 'value=i18n("是"),')
if new_line != line:
print("Replaced 'value=i18n(\"否\"),' with 'value=i18n(\"是\"),' based on the condition for SAVE SMALL WEIGHTS")
changes_made = True
line = new_line
temp_file.write(line)
second_previous_line = previous_line
previous_line = line
# After finished, we replace the original file with the temp one
import shutil
shutil.move(temp_file_path, file_path)
if changes_made:
print("Changes made and file saved successfully.")
else:
print("No changes were needed.")
# Define the repo path
repo_path = '/content/Retrieval-based-Voice-Conversion-WebUI'
def copy_all_files_in_directory(src_dir, dest_dir):
# Iterate over all files in source directory
for item in Path(src_dir).glob('*'):
if item.is_file():
# Copy each file to destination directory
shutil.copy(item, dest_dir)
else:
# If it's a directory, make a new directory in the destination and copy the files recursively
new_dest = Path(dest_dir) / item.name
new_dest.mkdir(exist_ok=True)
copy_all_files_in_directory(str(item), str(new_dest))
def clone_and_copy_repo(repo_path):
# New repository link
new_repo_link = "https://github.com/alexlnkp/Mangio-RVC-Tweaks/"
# Temporary path to clone the repository
temp_repo_path = "/content/temp_Mangio-RVC-Fork"
# New folder name
new_folder_name = "Mangio-RVC-Fork"
# Clone the latest code from the new repository to a temporary location
run_cmd(f"git clone {new_repo_link} {temp_repo_path}")
os.chdir(temp_repo_path)
run_cmd(f"git checkout ui-redesign-2")
# Edit the file here, before copying
#edit_file(f"{temp_repo_path}/infer-web.py")
# Copy all files from the cloned repository to the existing path
copy_all_files_in_directory(temp_repo_path, repo_path)
print(f"Copying all {new_folder_name} files from GitHub.")
# Change working directory back to /content/
os.chdir('/content/')
print("Changed path back to /content/")
# Remove the temporary cloned repository
shutil.rmtree(temp_repo_path)
# Call the function
clone_and_copy_repo(repo_path)
# Download the credentials file for RVC archive sheet
os.makedirs('/content/Retrieval-based-Voice-Conversion-WebUI/stats/', exist_ok=True)
run_cmd("wget -q https://cdn.discordapp.com/attachments/945486970883285045/1114717554481569802/peppy-generator-388800-07722f17a188.json -O /content/Retrieval-based-Voice-Conversion-WebUI/stats/peppy-generator-388800-07722f17a188.json")
# Forcefully delete any existing torchcrepe dependencies downloaded from an earlier run just in case
shutil.rmtree('/content/Retrieval-based-Voice-Conversion-WebUI/torchcrepe', ignore_errors=True)
shutil.rmtree('/content/torchcrepe', ignore_errors=True)
# Download the torchcrepe folder from the maxrmorrison/torchcrepe repository
run_cmd("git clone https://github.com/maxrmorrison/torchcrepe.git")
shutil.move('/content/torchcrepe/torchcrepe', '/content/Retrieval-based-Voice-Conversion-WebUI/')
shutil.rmtree('/content/torchcrepe', ignore_errors=True) # Delete the torchcrepe repository folder
# Change the current directory to /content/Retrieval-based-Voice-Conversion-WebUI
os.chdir('/content/Retrieval-based-Voice-Conversion-WebUI')
os.makedirs('pretrained', exist_ok=True)
os.makedirs('uvr5_weights', exist_ok=True)
def download_file(url, filepath):
response = requests.get(url, stream=True)
response.raise_for_status()
with open(filepath, "wb") as file:
for chunk in response.iter_content(chunk_size=8192):
if chunk:
file.write(chunk)
def download_pretrained_models():
pretrained_models = {
"pretrained": [
"D40k.pth",
"G40k.pth",
"f0D40k.pth",
"f0G40k.pth"
],
"pretrained_v2": [
"D40k.pth",
"G40k.pth",
"f0D40k.pth",
"f0G40k.pth",
"f0G48k.pth",
"f0D48k.pth"
],
"uvr5_weights": [
"HP2-人声vocals+非人声instrumentals.pth",
"HP5-主旋律人声vocals+其他instrumentals.pth"
]
}
base_url = "https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/"
base_path = "/content/Retrieval-based-Voice-Conversion-WebUI/"
# Calculate total number of files to download
total_files = sum(len(files) for files in pretrained_models.values()) + 1 # +1 for hubert_base.pt
with tqdm(total=total_files, desc="Downloading files") as pbar:
for folder, models in pretrained_models.items():
folder_path = os.path.join(base_path, folder)
os.makedirs(folder_path, exist_ok=True)
for model in models:
url = base_url + folder + "/" + model
filepath = os.path.join(folder_path, model)
download_file(url, filepath)
pbar.update()
# Download hubert_base.pt to the base path
hubert_url = base_url + "hubert_base.pt"
hubert_filepath = os.path.join(base_path, "hubert_base.pt")
download_file(hubert_url, hubert_filepath)
pbar.update()
def clone_repository(run_download):
with ThreadPoolExecutor(max_workers=2) as executor:
executor.submit(run_script)
if run_download:
executor.submit(download_pretrained_models)