Spaces:
Runtime error
Runtime error
File size: 1,653 Bytes
2de3774 |
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 |
import json
from pathlib import Path
import shutil
class PerformanceSettings:
DEFAULT_PERFORMANCE_FILE = Path("settings/performance.default")
PERFORMANCE_FILE = Path("settings/performance.json")
CUSTOM_PERFORMANCE = "Custom..."
default_settings = {
"custom_steps": 30,
"cfg": 8,
"sampler_name": "dpmpp_2m_sde_gpu",
"scheduler": "karras",
"clip_skip": 1,
}
def __init__(self):
self.performance_options = self.load_performance()
def load_performance(self):
default_data = self._load_data(self.DEFAULT_PERFORMANCE_FILE)
data = self._load_data(self.PERFORMANCE_FILE)
for name, default_settings in default_data.items():
if name not in data:
data[name] = default_settings
else:
existing_settings = data[name]
merged_settings = existing_settings | default_settings
data[name] = merged_settings
self._save_data(self.PERFORMANCE_FILE, data)
return data
def _load_data(self, file_path):
if not file_path.exists():
shutil.copy(self.DEFAULT_PERFORMANCE_FILE, file_path)
return json.load(open(file_path))
def _save_data(self, file_path, data):
with open(file_path, "w") as f:
json.dump(data, f, indent=2)
def save_performance(self, perf_options):
with open(self.PERFORMANCE_FILE, "w") as f:
json.dump(perf_options, f, indent=2)
self.performance_options = self.load_performance()
def get_perf_options(self, name):
return self.performance_options[name]
|