VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

replace unsafe eval() with ast

Browse Source
This commit is contained in:
Isotr0py
2023-03-10 13:44:16 +08:00
parent ab05be11d2
commit c4a596df9e
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
library/train_util.py
View File

@@ -1,6 +1,7 @@
# common functions for training # common functions for training
import argparse import argparse
import ast
import importlib import importlib
import json import json
import re import re
@@ -1886,11 +1887,11 @@ def get_scheduler_fix(args,optimizer: Optimizer):
if value[i].lower() == "true" or value[i].lower() == "false": if value[i].lower() == "true" or value[i].lower() == "false":
value[i] = (value[i].lower() == "true") value[i] = (value[i].lower() == "true")
else: else:
value[i] = eval(value[i]) # warning: not safe!!! value[i] = ast.literal_eval(value[i])
if len(value) == 1: if len(value) == 1:
value = value[0] value = value[0]
else: else:
value = tuple(value) # some may use list? value = list(value) # some may use list?
lr_scheduler_kwargs[key] = value lr_scheduler_kwargs[key] = value