Fixed paint your dream for python 3.9 (#114)

openvinotoolkit · Nov 29, 2024 · df7ff79 · df7ff79
1 parent 7ec2d94
commit df7ff79
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 94 deletions.
diff --git a/demos/paint_your_dreams_demo/last_known_config.txt b/demos/paint_your_dreams_demo/last_known_config.txt
diff --git a/demos/paint_your_dreams_demo/main.py b/demos/paint_your_dreams_demo/main.py
@@ -4,6 +4,7 @@
 import sys
 import time
 from pathlib import Path
+from typing import Optional
 
 import gradio as gr
 import numpy as np
@@ -21,8 +22,8 @@
 MAX_SEED = np.iinfo(np.int32).max
 MODEL_DIR = Path("model")
 
-ov_pipeline: OVLatentConsistencyModelPipeline | None = None
-safety_checker: StableDiffusionSafetyChecker | None = None
+ov_pipeline: Optional[OVLatentConsistencyModelPipeline] = None
+safety_checker: Optional[StableDiffusionSafetyChecker] = None
 
 try:
     safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker", local_files_only=True)
@@ -32,7 +33,7 @@
 ov_pipelines = {}
 
 stop_generating: bool = True
-hf_model_name: str | None = None
+hf_model_name: Optional[str] = None
 
 
 def get_available_devices() -> list[str]: