Skip to content

Add support for API and refactor#5

Open
vbaddi wants to merge 1 commit intomainfrom
enable_api_and_refactor
Open

Add support for API and refactor#5
vbaddi wants to merge 1 commit intomainfrom
enable_api_and_refactor

Conversation

@vbaddi
Copy link
Copy Markdown
Owner

@vbaddi vbaddi commented Feb 28, 2025

  • Refactor the codebase
  • Enable CLI and API both
from QEfficient.finetune import api
from QEfficient.finetune.configs import LoraConfig, TrainConfig

train_config = TrainConfig(model_name="meta-llama/Llama-3.2-1B", lr=5e-4, enable_ddp=False, device="qaic")
lora_config = LoraConfig(r=8, lora_alpha=32)
results = api.finetune(train_config, "/home/vbaddi/finetune_infra/efficient-transformers/lora_config.yaml")
print(results)

@vbaddi vbaddi force-pushed the enable_api_and_refactor branch from 2957d3b to a24730b Compare February 28, 2025 06:25
Signed-off-by: vbaddi <quic_vbaddi@quicinc.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant