"vscode:/vscode.git/clone" did not exist on "7c103cdad6807087ee378d0f9d8c2a5c957f3fb3"
Commit c128695d authored by chenpangpang's avatar chenpangpang
Browse files

feat: 修改app.py避免下载模型;dockerfile bug修复

parent 626ddc66
Pipeline #1586 failed with stages
in 0 seconds
...@@ -19,5 +19,6 @@ COPY chenyh/$IMAGE/Kwai-Kolors/Kolors /root/$IMAGE_UPPER/Kwai-Kolors/Kolors ...@@ -19,5 +19,6 @@ COPY chenyh/$IMAGE/Kwai-Kolors/Kolors /root/$IMAGE_UPPER/Kwai-Kolors/Kolors
COPY chenyh/$IMAGE/Kwai-Kolors/Kolors-IP-Adapter-Plus /root/$IMAGE_UPPER/Kwai-Kolors/Kolors-IP-Adapter-Plus COPY chenyh/$IMAGE/Kwai-Kolors/Kolors-IP-Adapter-Plus /root/$IMAGE_UPPER/Kwai-Kolors/Kolors-IP-Adapter-Plus
COPY --from=base /opt/conda/lib/python3.10/site-packages /opt/conda/lib/python3.10/site-packages COPY --from=base /opt/conda/lib/python3.10/site-packages /opt/conda/lib/python3.10/site-packages
COPY --from=base /root/$IMAGE/$IMAGE_UPPER /root/$IMAGE_UPPER COPY --from=base /root/$IMAGE/$IMAGE_UPPER /root/$IMAGE_UPPER
COPY --from=base /root/$IMAGE/启动器.ipynb /root/$IMAGE/start.sh /root/$IMAGE/assets /root/ COPY --from=base /root/$IMAGE/启动器.ipynb /root/$IMAGE/start.sh /root/
COPY --from=base /root/$IMAGE/assets /root/assets
...@@ -12,8 +12,8 @@ import gradio as gr ...@@ -12,8 +12,8 @@ import gradio as gr
import numpy as np import numpy as np
device = "cuda" device = "cuda"
ckpt_dir = snapshot_download(repo_id="Kwai-Kolors/Kolors") ckpt_dir = "Kwai-Kolors/Kolors"
ckpt_IPA_dir = snapshot_download(repo_id="Kwai-Kolors/Kolors-IP-Adapter-Plus") ckpt_IPA_dir = "Kwai-Kolors/Kolors-IP-Adapter-Plus"
text_encoder = ChatGLMModel.from_pretrained(f'{ckpt_dir}/text_encoder', torch_dtype=torch.float16).half().to(device) text_encoder = ChatGLMModel.from_pretrained(f'{ckpt_dir}/text_encoder', torch_dtype=torch.float16).half().to(device)
tokenizer = ChatGLMTokenizer.from_pretrained(f'{ckpt_dir}/text_encoder') tokenizer = ChatGLMTokenizer.from_pretrained(f'{ckpt_dir}/text_encoder')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment