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):
|