Skip to content

[Feature] Test case enhancement for Lora features #3414

@Fridge003

Description

@Fridge003

Checklist

Motivation

Currently there is only two test files for Lora features:
test/srt/models/test_lora.py and
test/srt/models/test_lora_backend.py
These two tests are only tested on llama models, thus not comprehensive.

Lora needs a series of well-organized tests, which can be similar to
test/srt/models/test_generation_models.py

Check List

check list copied from #3652:

  • Add backend test support for single adaptor, single prompt inference.
  • Add backend test support for single adaptor, batch prompts serving.
  • Add backend test support for multi-adaptor, same rank.
  • Add backend test support for multi-adaptor, different rank.
  • Add backend test support for adaptor with Embedding and Lm_head layer weights.

Related PRs

#3652 #4239 #4925

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions