diff --git a/docs/Extensions.md b/docs/Extensions.md
index c7d1aa36..4e59e855 100644
--- a/docs/Extensions.md
+++ b/docs/Extensions.md
@@ -111,14 +111,16 @@ functions are declared in the same order that they are called at
 generation time.
 """
 
+import gradio as gr
 import torch
-from modules import chat
+from transformers import LogitsProcessor
+
+from modules import chat, shared
 from modules.text_generation import (
     decode,
     encode,
     generate_reply,
 )
-from transformers import LogitsProcessor
 
 params = {
     "display_name": "Example Extension",
diff --git a/extensions/example/script.py b/extensions/example/script.py
index 669749c0..b4db7102 100644
--- a/extensions/example/script.py
+++ b/extensions/example/script.py
@@ -7,14 +7,16 @@ functions are declared in the same order that they are called at
 generation time.
 """
 
+import gradio as gr
 import torch
-from modules import chat
+from transformers import LogitsProcessor
+
+from modules import chat, shared
 from modules.text_generation import (
     decode,
     encode,
     generate_reply,
 )
-from transformers import LogitsProcessor
 
 params = {
     "display_name": "Example Extension",
diff --git a/extensions/sd_api_pictures/script.py b/extensions/sd_api_pictures/script.py
index 78488cd0..88a0d940 100644
--- a/extensions/sd_api_pictures/script.py
+++ b/extensions/sd_api_pictures/script.py
@@ -332,8 +332,8 @@ def ui():
             negative_prompt = gr.Textbox(placeholder=params['negative_prompt'], value=params['negative_prompt'], label='Negative Prompt')
             with gr.Row():
                 with gr.Column():
-                    width = gr.Slider(256, 768, value=params['width'], step=64, label='Width')
-                    height = gr.Slider(256, 768, value=params['height'], step=64, label='Height')
+                    width = gr.Slider(64, 2048, value=params['width'], step=64, label='Width')
+                    height = gr.Slider(64, 2048, value=params['height'], step=64, label='Height')
                 with gr.Column(variant="compact", elem_id="sampler_col"):
                     with gr.Row(elem_id="sampler_row"):
                         sampler_name = gr.Dropdown(value=params['sampler_name'], label='Sampling method', elem_id="sampler_box")