Skip to content

ZJU-ACES-ISE/chatunitest-models

Repository files navigation

ChatUniTest Models

Here we provide finetuned models for Java test generation tasks based on Code Llama.

Models

Model Base Model parameters Model Weights
codellama-testgen CodeLlama 7B zzzghttt/CodeLlama-7b-Test-Instruct-lora
codellama-testgen2 CodeLlama 7B zzzghttt/TestGen2-lora

Usage

  1. Start the model server
python model_server.py
  1. Run a completion or generation route
python chatunitest_completion_route.py
  1. Then you can make requests by configure the url in ChatUniTest, see example request script in completion_example.py.

Details

Fine-Tuning Method

The fine-tuning process of codellama-testgen2. finetune

Releases

No releases published

Packages

No packages published

Languages