nodes/kijai-ComfyUI-Florence2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

clarify print

Browse Source
This commit is contained in:
kijai
2025-01-12 19:55:45 +02:00
parent 47b554708e
commit dffd12506d
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+2 -2
View File
@@ -96,7 +96,7 @@ class DownloadAndLoadFlorence2Model:
local_dir=model_path,
local_dir_use_symlinks=False)
print(f"using {attention} for attention")
print(f"Florence2 using {attention} for attention")
with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports): #workaround for unnecessary flash_attn requirement
model = AutoModelForCausalLM.from_pretrained(model_path, attn_implementation=attention, device_map=device, torch_dtype=dtype,trust_remote_code=True)
processor = AutoProcessor.from_pretrained(model_path, trust_remote_code=True)
@@ -172,7 +172,7 @@ class Florence2ModelLoader:
dtype = {"bf16": torch.bfloat16, "fp16": torch.float16, "fp32": torch.float32}[precision]
model_path = Path(folder_paths.models_dir, "LLM", model)
print(f"Loading model from {model_path}")
print(f"using {attention} for attention")
print(f"Florence2 using {attention} for attention")
with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports): #workaround for unnecessary flash_attn requirement
model = AutoModelForCausalLM.from_pretrained(model_path, attn_implementation=attention, device_map=device, torch_dtype=dtype,trust_remote_code=True)
processor = AutoProcessor.from_pretrained(model_path, trust_remote_code=True)
pyproject.toml
+1 -1
View File
@@ -1,7 +1,7 @@
[project]
name = "comfyui-florence2"
description = "Nodes to use Florence2 VLM for image vision tasks: object detection, captioning, segmentation and ocr"
version = "1.0.2"
version = "1.0.3"
license = "MIT"
dependencies = ["transformers>=4.38.0"]