Spaces:
Sleeping
Sleeping
Update load_models_utils.py
Browse files- load_models_utils.py +0 -5
load_models_utils.py
CHANGED
@@ -5,18 +5,13 @@ from utils import PhotoMakerStableDiffusionXLPipeline
|
|
5 |
import os
|
6 |
|
7 |
def get_models_dict():
|
8 |
-
# 打开并读取YAML文件
|
9 |
with open('config/models.yaml', 'r') as stream:
|
10 |
try:
|
11 |
-
# 解析YAML文件内容
|
12 |
data = yaml.safe_load(stream)
|
13 |
-
|
14 |
-
# 此时 'data' 是一个Python字典,里面包含了YAML文件的所有数据
|
15 |
print(data)
|
16 |
return data
|
17 |
|
18 |
except yaml.YAMLError as exc:
|
19 |
-
# 如果在解析过程中发生了错误,打印异常信息
|
20 |
print(exc)
|
21 |
|
22 |
def load_models(model_info,device,photomaker_path):
|
|
|
5 |
import os
|
6 |
|
7 |
def get_models_dict():
|
|
|
8 |
with open('config/models.yaml', 'r') as stream:
|
9 |
try:
|
|
|
10 |
data = yaml.safe_load(stream)
|
|
|
|
|
11 |
print(data)
|
12 |
return data
|
13 |
|
14 |
except yaml.YAMLError as exc:
|
|
|
15 |
print(exc)
|
16 |
|
17 |
def load_models(model_info,device,photomaker_path):
|