Skip to content

Question about Inference #4

@thaomaidang

Description

@thaomaidang

Dear Authors,

First of all, thank you for your wonderful work. I have encountered several issues when running your inference code:

  1. I have an issue regarding missing modules: "model_transformer", "pos_embed", and "utils".
    For example: from model_transformer import LayerNormFp32, LayerNorm, QuickGELU, VisionTransformer - line 7 in KEP_model.py

  2. I tried to find the model path in your GitHub repository, but there is no model path (line 6-7 in inference/easy_inference.py)
    model_name = 'KEP_CTP_OpenPath'
    model_bin_path = 'Path/to/model/root/' + model_name

  3. There is no 'pytorch_model.bin' - line 314 in KEP_model.py
    checkpoint = torch.load(model_path + '/pytorch_model.bin', map_location='cpu')

Thank you in advance, I really appreciate your work.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions