mirror of
https://github.com/tcsenpai/pensieve.git
synced 2025-06-06 03:05:25 +00:00
chore: remove unused import
This commit is contained in:
parent
033417438b
commit
9d968f61f3
@ -9,12 +9,9 @@ import logging
|
|||||||
import uvicorn
|
import uvicorn
|
||||||
import os
|
import os
|
||||||
import io
|
import io
|
||||||
import torch
|
|
||||||
from transformers import AutoModelForCausalLM, AutoProcessor
|
|
||||||
|
|
||||||
from unittest.mock import patch
|
from unittest.mock import patch
|
||||||
from transformers.dynamic_module_utils import get_imports
|
from transformers.dynamic_module_utils import get_imports
|
||||||
from modelscope import snapshot_download
|
|
||||||
|
|
||||||
def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
|
def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
|
||||||
if not str(filename).endswith("modeling_florence2.py"):
|
if not str(filename).endswith("modeling_florence2.py"):
|
||||||
|
Loading…
x
Reference in New Issue
Block a user