Skip to content

Commit 1eb45cd

Browse files
authored
Fix ckpt in docs (#41659)
* fix ckpt in docs * fix config ckpt
1 parent 354567d commit 1eb45cd

File tree

4 files changed

+10
-10
lines changed

4 files changed

+10
-10
lines changed

docs/source/en/model_doc/florence2.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -70,8 +70,8 @@ from transformers import AutoProcessor, Florence2ForConditionalGeneration
7070
url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/car.jpg?download=true"
7171
image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
7272

73-
model = Florence2ForConditionalGeneration.from_pretrained("microsoft/Florence-2-base", dtype=torch.bfloat16, device_map="auto")
74-
processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base")
73+
model = Florence2ForConditionalGeneration.from_pretrained("florence-community/Florence-2-base", dtype=torch.bfloat16, device_map="auto")
74+
processor = AutoProcessor.from_pretrained("florence-community/Florence-2-base")
7575

7676
task_prompt = "<OD>"
7777
inputs = processor(text=task_prompt, images=image, return_tensors="pt").to(model.device)
@@ -105,12 +105,12 @@ from transformers import AutoProcessor, Florence2ForConditionalGeneration, BitsA
105105
quantization_config = BitsAndBytesConfig(load_in_4bit=True)
106106

107107
model = Florence2ForConditionalGeneration.from_pretrained(
108-
"microsoft/Florence-2-large",
108+
"florence-community/Florence-2-base",
109109
dtype=torch.bfloat16,
110110
device_map="auto",
111111
quantization_config=quantization_config
112112
)
113-
processor = AutoProcessor.from_pretrained("microsoft/Florence-2-large")
113+
processor = AutoProcessor.from_pretrained("florence-community/Florence-2-base")
114114

115115
url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/car.jpg?download=true"
116116
image = Image.open(requests.get(url, stream=True).raw).convert("RGB")

src/transformers/models/florence2/configuration_florence2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ class Florence2Config(PreTrainedConfig):
140140
Florence-2 model according to the specified arguments, defining the model architecture.
141141
142142
Instantiating a configuration with the defaults will yield a similar configuration to that of the Florence-2
143-
[microsoft/Florence-2-base](https://huggingface.co/microsoft/Florence-2-base) architecture.
143+
[florence-community/Florence-2-base](https://huggingface.co/florence-community/Florence-2-base) architecture.
144144
145145
Configuration objects inherit from [`PreTrainedConfig`] and can be used to control the model outputs. Read the
146146
documentation from [`PreTrainedConfig`] for more information.

src/transformers/models/florence2/modeling_florence2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -884,8 +884,8 @@ def forward(
884884
>>> import requests
885885
>>> from transformers import AutoProcessor, Florence2ForConditionalGeneration
886886
887-
>>> model = Florence2ForConditionalGeneration.from_pretrained("microsoft/Florence-2-large")
888-
>>> processor = AutoProcessor.from_pretrained("microsoft/Florence-2-large")
887+
>>> model = Florence2ForConditionalGeneration.from_pretrained("florence-community/Florence-2-large")
888+
>>> processor = AutoProcessor.from_pretrained("florence-community/Florence-2-large")
889889
890890
>>> prompt = "<CAPTION>"
891891
>>> url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/car.jpg"

src/transformers/models/florence2/modular_florence2.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ class Florence2Config(PreTrainedConfig):
160160
Florence-2 model according to the specified arguments, defining the model architecture.
161161
162162
Instantiating a configuration with the defaults will yield a similar configuration to that of the Florence-2
163-
[microsoft/Florence-2-base](https://huggingface.co/microsoft/Florence-2-base) architecture.
163+
[florence-community/Florence-2-base](https://huggingface.co/florence-community/Florence-2-base) architecture.
164164
165165
Configuration objects inherit from [`PreTrainedConfig`] and can be used to control the model outputs. Read the
166166
documentation from [`PreTrainedConfig`] for more information.
@@ -1674,8 +1674,8 @@ def forward(
16741674
>>> import requests
16751675
>>> from transformers import AutoProcessor, Florence2ForConditionalGeneration
16761676
1677-
>>> model = Florence2ForConditionalGeneration.from_pretrained("microsoft/Florence-2-large")
1678-
>>> processor = AutoProcessor.from_pretrained("microsoft/Florence-2-large")
1677+
>>> model = Florence2ForConditionalGeneration.from_pretrained("florence-community/Florence-2-large")
1678+
>>> processor = AutoProcessor.from_pretrained("florence-community/Florence-2-large")
16791679
16801680
>>> prompt = "<CAPTION>"
16811681
>>> url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/car.jpg"

0 commit comments

Comments
 (0)