fix style iwht black (#2769)

Author: eaidova

notebooks/image-to-image-genai/gradio_helper.py

@@ -15,7 +15,19 @@
 
 
 def make_demo(pipeline, generator_cls, image_to_tensor):
-    def infer(input_image, prompt, negative_prompt, seed, strength, randomize_seed, num_inference_steps, use_custom_size, height, width, progress=gr.Progress(track_tqdm=True)):
+    def infer(
+        input_image,
+        prompt,
+        negative_prompt,
+        seed,
+        strength,
+        randomize_seed,
+        num_inference_steps,
+        use_custom_size,
+        height,
+        width,
+        progress=gr.Progress(track_tqdm=True),
+    ):
         if randomize_seed:
             seed = np.random.randint(0, MAX_SEED)
 
@@ -30,12 +42,12 @@ def callback(step, num_steps, latent):
             pbar.update(1)
             sys.stdout.flush()
             return False
+
         additional_args = {}
 
         if use_custom_size:
             additional_args = {"height": height, "width": width}
 
-
         image_tensor = pipeline.generate(
             prompt,
             init_image_tensor,

notebooks/image-to-image-genai/image-to-image-genai.ipynb

@@ -499,7 +499,14 @@
     "\n",
     "\n",
     "image_tensor = pipe.generate(\n",
-    "    prompt, init_image_tensor, negative_prompt=negative_prompt, strength=strength, num_inference_steps=20, num_images_per_prompt=1, generator=random_generator, callback=callback\n",
+    "    prompt,\n",
+    "    init_image_tensor,\n",
+    "    negative_prompt=negative_prompt,\n",
+    "    strength=strength,\n",
+    "    num_inference_steps=20,\n",
+    "    num_images_per_prompt=1,\n",
+    "    generator=random_generator,\n",
+    "    callback=callback,\n",
     ")\n",
     "\n",
     "pbar.close()\n",

supplementary_materials/notebooks/fastdraft-deepseek/fastdraft_deepseek.ipynb

@@ -102,7 +102,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "\n",
     "if not model_dir.exists():\n",
     "    ! optimum-cli export openvino --model $model_id --task text-generation-with-past --weight-format int4 $model_dir\n",
     "\n",
@@ -407,9 +406,10 @@
     "\n",
     "print(\"Loading prompts...\")\n",
     "import json\n",
-    "f= open('prompts.json')\n",
+    "\n",
+    "f = open(\"prompts.json\")\n",
     "prompts = json.load(f)\n",
-    "prompts = [[{\"role\": \"user\", \"content\": p }] for p in prompts]\n",
+    "prompts = [[{\"role\": \"user\", \"content\": p}] for p in prompts]\n",
     "\n",
     "# We will first do a short warmup to the model so the time measurement will not include the warmup overhead.\n",
     "generation_config.max_new_tokens = 8\n",
@@ -470,9 +470,10 @@
     "\n",
     "print(\"Loading prompts...\")\n",
     "import json\n",
-    "f= open('prompts.json')\n",
+    "\n",
+    "f = open(\"prompts.json\")\n",
     "prompts = json.load(f)\n",
-    "prompts = [[{\"role\": \"user\", \"content\": p }] for p in prompts]\n",
+    "prompts = [[{\"role\": \"user\", \"content\": p}] for p in prompts]\n",
     "\n",
     "# Define scheduler\n",
     "scheduler_config = ov_genai.SchedulerConfig()\n",