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-
docker run -it \
--shm-size 60g \
--network=host \
--name Spark-X1 \
--name audiofly \
--privileged \
--device=/dev/kfd \
--device=/dev/dri \
......
......@@ -3,10 +3,10 @@ import torch
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"])
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.eval()
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