diff --git a/src/llama_stack/providers/inline/post_training/huggingface/recipes/finetune_single_device.py b/src/llama_stack/providers/inline/post_training/huggingface/recipes/finetune_single_device.py index 61af0f9e3..4506a2351 100644 --- a/src/llama_stack/providers/inline/post_training/huggingface/recipes/finetune_single_device.py +++ b/src/llama_stack/providers/inline/post_training/huggingface/recipes/finetune_single_device.py @@ -14,7 +14,6 @@ import torch from datasets import Dataset from peft import LoraConfig from transformers import ( - AutoModelForCausalLM, AutoTokenizer, ) from trl import SFTConfig, SFTTrainer diff --git a/src/llama_stack/testing/api_recorder.py b/src/llama_stack/testing/api_recorder.py index c66606a56..dfa541872 100644 --- a/src/llama_stack/testing/api_recorder.py +++ b/src/llama_stack/testing/api_recorder.py @@ -10,7 +10,7 @@ import hashlib import json import os import re -from collections.abc import Callable, Generator, Sequence +from collections.abc import Callable, Generator from contextlib import contextmanager from enum import StrEnum from pathlib import Path