Merge pull request #30 from EtienneDosSantos/29-timeouts-and-no-hip-g…

…pu-errors 29 timeouts and no hip gpu errors
IndigoDosSantos · Mar 17, 2024 · c56d43b · c56d43b
2 parents 9de7dbf + 750fea6
commit c56d43b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -22,9 +22,9 @@ onnx2torch>=1.5.13
 onnxruntime>=1.16.3
 opencv-python>=4.8.1.78
 peft
-torch>=2.1.2+cu118
+torch==2.2.1+cu121
 torchtools @ git+https://github.com/pabloppp/pytorch-tools
-torchvision>=0.16.2+cu118
+torchvision==0.17.1+cu121
 tqdm>=4.66.1
 transformers>=4.30.0
 wandb>=0.16.2

diff --git a/run.py b/run.py
@@ -128,7 +128,7 @@ def generate_images(prompt, height, width, negative_prompt, guidance_scale, num_
 
 # UI Layout putting Configurator blocks inside a function for clarity.
 def configure_ui():
-    with gr.Blocks(theme=gr.themes.Soft()) as demo: # Change to your desired theme
+    with gr.Blocks(theme=gr.themes.Soft(), analytics_enabled=False) as demo: # Change to your desired theme
         gr.HTML("""
         <link rel="preconnect" href="https://fonts.googleapis.com">
         <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>