Skip to content

Request to Release Demo Checkpoint & Clarification on Public Model Versions #4

Description

@sgallon-rin

Hi everyone, and thanks for all the hard work on this repo!

When I ran demo.sh, I hit this error:

.../StickerConv/model$ bash run_script/demo.sh
.......
Traceback (most recent call last):
  File "demo.py", line 135, in <module>
    main(args)
  File "demo.py", line 49, in main
    agent = ConversationalAgent(config)
  File ".../StickerConv/model/demo/agent.py", line 19, in __init__
    self._load_checkpoint(self.checkpoint_path)
  File ".../StickerConv/model/demo/agent.py", line 77, in _load_checkpoint
    raise ValueError("checkpoint path is invalid!")
ValueError: checkpoint path is invalid!

It looks like the checkpoint referenced in demo/config/demo.yaml isn’t publicly available:

model:
    checkpoint: "outputs/joint_learning/checkpoint_final.pth"

Could you share the trained final checkpoint?


Also, in pegs/configs/common/pegs.yaml, pre-trained model paths like:

model:
    llama_pretrained_model_path_or_name: "/datas/huggingface/vicuna-7b/"
    blip2_pretrained_model_path_or_name: "/datas/huggingface/blip2-opt-2.7b/"
    stable_diffusion_pretrained_model_path_or_name: "/datas/huggingface/cuteyukimix_SD1.5/diffusers/"

Could you point me to the exact public versions (Hugging Face repos, download URLs) that correspond to each of these entries?

Thanks in advance!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions