fix

huggingface · Apr 12, 2024 · 974d5a5 · 974d5a5
1 parent 3352701
commit 974d5a5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/.github/workflows/doctests.yml b/.github/workflows/doctests.yml
@@ -3,7 +3,7 @@ name: Doctests
 on:
   push:
     branches:
-      - run_doctest*
+      - fix_doctest_based_on_refactor_doctest
   repository_dispatch:
   schedule:
     - cron: "17 2 * * *"

diff --git a/src/transformers/models/fuyu/modeling_fuyu.py b/src/transformers/models/fuyu/modeling_fuyu.py
@@ -249,7 +249,7 @@ def forward(
         >>> inputs = processor(text=prompt, images=image, return_tensors="pt")
         >>> outputs = model(**inputs)
 
-        >>> generated_ids = model.generate(**model_inputs, max_new_tokens=7)
+        >>> generated_ids = model.generate(**inputs, max_new_tokens=7)
         >>> generation_text = processor.batch_decode(generated_ids, skip_special_tokens=True)
         >>> print(generation_text)
         'A bus parked on the side of a road.'

diff --git a/utils/split_doctest_jobs.py b/utils/split_doctest_jobs.py
@@ -85,6 +85,10 @@
         end = start + num_jobs_per_splits + (1 if idx < num_jobs % args.num_splits else 0)
         file_directory_splits.append(sorted_file_dirs[start:end])
 
+    file_directory_splits = [
+        ["src/transformers/models/fuyu"],
+    ]
+
     if args.only_return_keys:
         print(file_directory_splits)
     else: