From 21549b7bfe9af0d3a1274d7aca5aa140b93e889d Mon Sep 17 00:00:00 2001 From: Gang Wen Date: Tue, 7 Apr 2026 19:27:39 +0200 Subject: [PATCH] Fix florence2_models imports: add __init__.py and use relative imports The florence2_models directory needs an __init__.py to be a proper Python package, and imports from py/florence2_ultra.py need to use relative imports (..florence2_models) since it's a sibling subpackage. Co-Authored-By: Claude Opus 4.6 --- florence2_models/__init__.py | 0 py/florence2_ultra.py | 4 ++-- 2 files changed, 2 insertions(+), 2 deletions(-) create mode 100644 florence2_models/__init__.py diff --git a/florence2_models/__init__.py b/florence2_models/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/py/florence2_ultra.py b/py/florence2_ultra.py index 4048a1c..6d87d3a 100644 --- a/py/florence2_ultra.py +++ b/py/florence2_ultra.py @@ -48,9 +48,9 @@ def fixed_get_imports(filename) -> list[str]: def _load_model_v5(model_path, attention, dtype): """Load Florence2 model for transformers >= 5.0.0""" log(f"[DEBUG] _load_model_v5 called with model_path={model_path}, attention={attention}, dtype={dtype}") - from florence2_models.modeling_florence2 import Florence2ForConditionalGeneration, Florence2Config + from ..florence2_models.modeling_florence2 import Florence2ForConditionalGeneration, Florence2Config from transformers import CLIPImageProcessor, BartTokenizerFast - from florence2_models.processing_florence2 import Florence2Processor + from ..florence2_models.processing_florence2 import Florence2Processor from accelerate import init_empty_weights from accelerate.utils import set_module_tensor_to_device from comfy.utils import load_torch_file