Spaces:
Running
on
Zero
Running
on
Zero
刘虹雨
commited on
Commit
·
bbd0d54
1
Parent(s):
23af8a2
update code
Browse files
app.py
CHANGED
@@ -269,10 +269,30 @@ def prepare_working_dir(dir, style):
|
|
269 |
|
270 |
|
271 |
def launch_pretrained():
|
272 |
-
|
273 |
-
|
274 |
-
|
275 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
276 |
|
277 |
|
278 |
def prepare_image_list(img_dir, selected_img):
|
|
|
269 |
|
270 |
|
271 |
def launch_pretrained():
|
272 |
+
|
273 |
+
from huggingface_hub import snapshot_download
|
274 |
+
|
275 |
+
snapshot_download(
|
276 |
+
repo_id="KumaPower/AvatarArtist",
|
277 |
+
repo_type="model",
|
278 |
+
local_dir="./pretrained_model",
|
279 |
+
local_dir_use_symlinks=False
|
280 |
+
)
|
281 |
+
|
282 |
+
snapshot_download(
|
283 |
+
repo_id="stabilityai/stable-diffusion-2-1-base",
|
284 |
+
repo_type="model",
|
285 |
+
local_dir="./pretrained_model/sd21",
|
286 |
+
local_dir_use_symlinks=False
|
287 |
+
)
|
288 |
+
|
289 |
+
# 下载 CrucibleAI/ControlNetMediaPipeFace 的所有文件
|
290 |
+
snapshot_download(
|
291 |
+
repo_id="CrucibleAI/ControlNetMediaPipeFace",
|
292 |
+
repo_type="model",
|
293 |
+
local_dir="./pretrained_model/control",
|
294 |
+
local_dir_use_symlinks=False
|
295 |
+
)
|
296 |
|
297 |
|
298 |
def prepare_image_list(img_dir, selected_img):
|