Added Style embeddings

main_inference.py

@@ -14,7 +14,7 @@ import os
 from utils import color_loss,latents_to_pil,pil_to_latent,sketch_loss
 # Set device
 torch_device =  "cpu"
-# Load the Model
+
 vae = AutoencoderKL.from_pretrained("CompVis/stable-diffusion-v1-4", subfolder="vae")
 
 # Load the tokenizer and text encoder to tokenize and encode the text.
@@ -226,4 +226,4 @@ def generate_noised_version_of_image(pil_image):
 #     #pil_image = generate_image(prompt,color_postprocessing,color_loss_scale)
 #     #pil_image = generate_noised_version_of_image(Image.open('output.png').resize((512, 512)))
 #     pil_image.save("output1.png")
-    
+    

requirements.txt

@@ -1,13 +1,32 @@
-diffusers==0.21.4
-gradio==3.49.0
-ipython==8.15.0
-matplotlib==3.8.0
-numpy==1.26.1
-opencv_python==4.8.1.78
-Pillow==10.1.0
-Pillow==10.1.0
-torch==2.1.0
-torchvision==0.16.0
-tqdm==4.66.1
-transformers==4.25.1
-scipy==1.11.1
+albumentations==1.3.1
+annoy==1.17.3
+backports.functools-lru-cache==1.6.5
+black==23.7.0
+brotlipy==0.7.0
+clip==1.0
+clip-rerank==0.1.0
+conda-content-trust==0.1.3
+conda-libmamba-solver==23.5.0
+datasets==2.14.4
+diffusers==0.2.4
+gensim==4.3.0
+grad-cam==1.4.8
+gradio==3.45.2
+ipykernel==6.25.1
+kaggle==1.5.16
+lightning==2.0.6
+lightning-bolts==0.7.0
+lmdb==1.4.1
+mkl-fft==1.3.8
+mkl-random==1.2.4
+open-clip-torch==2.20.0
+pip-chill==1.0.3
+pixelator==1.1.0
+pre-commit==3.3.3
+pydeprecate==0.3.2
+pysocks==1.7.1
+ruamel.yaml==0.17.21
+seaborn==0.12.2
+sentence-transformers==2.2.2
+torch-lr-finder==0.2.1
+torchinfo==1.8.0