Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-01-13 13:52:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

Added a 'clean' flag to not resume download.

Browse Source
This commit is contained in:
Nikita Skakun 2023-03-30 00:42:19 -07:00
parent e17af59261
commit 8c590c2362
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
download-model.py
View File

@ -24,12 +24,13 @@ parser.add_argument('--branch', type=str, default='main', help='Name of the Git
parser.add_argument('--threads', type=int, default=1, help='Number of files to download simultaneously.') parser.add_argument('--threads', type=int, default=1, help='Number of files to download simultaneously.')
parser.add_argument('--text-only', action='store_true', help='Only download text files (txt/json).') parser.add_argument('--text-only', action='store_true', help='Only download text files (txt/json).')
parser.add_argument('--output', type=str, default=None, help='The folder where the model should be saved.') parser.add_argument('--output', type=str, default=None, help='The folder where the model should be saved.')
parser.add_argument('--clean', action='store_true', help='Does not resume the previous download.')
args = parser.parse_args() args = parser.parse_args()
def get_file(url, output_folder): def get_file(url, output_folder):
filename = Path(url.rsplit('/', 1)[1]) filename = Path(url.rsplit('/', 1)[1])
output_path = output_folder / filename output_path = output_folder / filename
if output_path.exists(): if output_path.exists() and not args.clean:
# Check if the file has already been downloaded completely # Check if the file has already been downloaded completely
r = requests.head(url) r = requests.head(url)
total_size = int(r.headers.get('content-length', 0)) total_size = int(r.headers.get('content-length', 0))