mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-11 19:56:03 +00:00
pre-commit
This commit is contained in:
parent
36c57e3b5b
commit
69fb43b7ee
2 changed files with 1 additions and 2 deletions
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue