update codes

c4c64ae7 · dengjb · 6eba53ba · c4c64ae7 · c4c64ae7
Commit c4c64ae7 authored Jun 17, 2024 by dengjb
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

README.md README.md +1 -1

docker/Dockerfile docker/Dockerfile +1 -0

No files found.
--- a/README.md
+++ b/README.md
@@ -68,7 +68,7 @@ export HF_ENDPOINT=https://hf-mirror.com
 ## 推理
 1、基于Huggingface模型进行推理.<br>
-2、运行代码后模型会自动下载 默认需存放至models文件夹中<br>
+2、运行代码后ic-light模型会自动下载 默认存放至models文件夹中<br>
 3、也可自行更改 inference.py文件中的 model_name 参数<br>
 4、由于模型使用stablediffusionapi/realistic-vision-v51作为生成模型，所以需要进行下载<br>
 [stablediffusionapi/realistic-vision-v51下载地址](https://huggingface.co/stablediffusionapi/realistic-vision-v51)

--- a/docker/Dockerfile
+++ b/docker/Dockerfile
+FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk24.04-py310
\ No newline at end of file