From 9257d0e4a1bdf61ec45ebfa24e02cf7c6c70998a Mon Sep 17 00:00:00 2001 From: raspawar Date: Tue, 25 Mar 2025 20:34:51 +0530 Subject: [PATCH] revert change in lora api definition --- llama_stack/apis/post_training/post_training.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/llama_stack/apis/post_training/post_training.py b/llama_stack/apis/post_training/post_training.py index 57ad27072..d49668e23 100644 --- a/llama_stack/apis/post_training/post_training.py +++ b/llama_stack/apis/post_training/post_training.py @@ -8,7 +8,7 @@ from datetime import datetime from enum import Enum from typing import Any, Dict, List, Literal, Optional, Protocol, Union -from pydantic import BaseModel, ConfigDict, Field +from pydantic import BaseModel, Field from typing_extensions import Annotated from llama_stack.apis.common.content_types import URL @@ -71,7 +71,6 @@ class TrainingConfig(BaseModel): @json_schema_type class LoraFinetuningConfig(BaseModel): - model_config = ConfigDict(extra="allow") type: Literal["LoRA"] = "LoRA" lora_attn_modules: List[str] apply_lora_to_mlp: bool