Skip to content

Commit

Permalink
[lora] Log adapter name in inference
Browse files Browse the repository at this point in the history
  • Loading branch information
xyang16 committed Nov 21, 2024
1 parent 1be5f9a commit 013d47e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion engines/python/setup/djl_python/input_parser.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@
# BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, express or implied. See the License for
# the specific language governing permissions and limitations under the License.
import logging
import os
from dataclasses import dataclass, field
from typing import List, Dict, Optional, Callable

Expand Down Expand Up @@ -244,6 +243,7 @@ def _fetch_adapters_from_input(input_map: dict, input_item: Input):
if not isinstance(adapters_per_item, list):
adapters_per_item = [adapters_per_item]

logging.debug(f"Using adapter {adapters_per_item}")
return adapters_per_item


Expand Down

0 comments on commit 013d47e

Please sign in to comment.