Commit 4ad2e834 authored by weishb's avatar weishb
Browse files

更新镜像名,修改run.py敏感信息

parent a19a32cd
...@@ -24,7 +24,7 @@ harbor.sourcefind.cn:5443/dcu/admin/base/custom:vllm0.18.1-ubuntu22.04-dtk26.04- ...@@ -24,7 +24,7 @@ harbor.sourcefind.cn:5443/dcu/admin/base/custom:vllm0.18.1-ubuntu22.04-dtk26.04-
docker run -it \ docker run -it \
--shm-size 60g \ --shm-size 60g \
--network=host \ --network=host \
--name Spark-X1 \ --name audiofly \
--privileged \ --privileged \
--device=/dev/kfd \ --device=/dev/kfd \
--device=/dev/dri \ --device=/dev/dri \
......
...@@ -3,10 +3,10 @@ import torch ...@@ -3,10 +3,10 @@ import torch
from ldm.utils.util import instantiate_from_config from ldm.utils.util import instantiate_from_config
configs = yaml.load(open('/public/home/weishb/iflytek/AudioFly/config/config.yaml', "r"), Loader=yaml.FullLoader) configs = yaml.load(open('/path/to/model/iflytek/AudioFly/config/config.yaml', "r"), Loader=yaml.FullLoader)
model = instantiate_from_config(configs["model"]) model = instantiate_from_config(configs["model"])
checkpoint = torch.load('/public/home/weishb/iflytek/AudioFly/models/ldm/model.ckpt') checkpoint = torch.load('/path/to/model/iflytek/AudioFly/models/ldm/model.ckpt')
model.load_state_dict(checkpoint, strict=False) model.load_state_dict(checkpoint, strict=False)
model.eval() model.eval()
model = model.cuda() model = model.cuda()
......
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