pre-commit

This commit is contained in:
Ashwin Bharambe 2025-10-28 06:16:49 -07:00
parent 36c57e3b5b
commit 69fb43b7ee
2 changed files with 1 additions and 2 deletions

View file

@ -14,7 +14,6 @@ import torch
from datasets import Dataset from datasets import Dataset
from peft import LoraConfig from peft import LoraConfig
from transformers import ( from transformers import (
AutoModelForCausalLM,
AutoTokenizer, AutoTokenizer,
) )
from trl import SFTConfig, SFTTrainer from trl import SFTConfig, SFTTrainer

View file

@ -10,7 +10,7 @@ import hashlib
import json import json
import os import os
import re import re
from collections.abc import Callable, Generator, Sequence from collections.abc import Callable, Generator
from contextlib import contextmanager from contextlib import contextmanager
from enum import StrEnum from enum import StrEnum
from pathlib import Path from pathlib import Path