Delete extensions/sd_webui_realtime_lcm_canvas

extensions/sd_webui_realtime_lcm_canvas/README.md

@@ -1,17 +0,0 @@
-# Realtime-LCM-Canvas
-
-Extension for [webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui). This is a real-time sketch to image demo using LCM and the [gradio library](https://github.com/gradio-app/gradio). 
-If you're not familiar with LCM, read about it here - [article on Hugging Face](https://huggingface.co/blog/lcm_lora).
-
-
-![](preview.png)
-
-Find the UI for realtime-lcm-canvas in the Extras tab after installing the extension.
-
-# Installation
-
-Install from webui's Extensions tab.
-
-# Credits
-
-* flowty-realtime-lcm-canvas library that does all the work: https://github.com/flowtyone/flowty-realtime-lcm-canvas

extensions/sd_webui_realtime_lcm_canvas/install.py

@@ -1,7 +0,0 @@
-import launch
-
-if not launch.is_installed("diffusers"):
-    try:
-        launch.run_pip("install diffusers==0.24.0", "requirements for diffusers")
-    except Exception:
-        print("Can't install diffusers. Please follow the readme to install manually")

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/blobs/21fbcd7923fb852577f1ade3e96282b689aa826a

@@ -1,34 +0,0 @@
-{
-  "_class_name": "StableDiffusionPipeline",
-  "_diffusers_version": "0.21.0.dev0",
-  "_name_or_path": "lykon-models/dreamshaper-7",
-  "feature_extractor": [
-    "transformers",
-    "CLIPFeatureExtractor"
-  ],
-  "requires_safety_checker": true,
-  "safety_checker": [
-    "stable_diffusion",
-    "StableDiffusionSafetyChecker"
-  ],
-  "scheduler": [
-    "diffusers",
-    "DEISMultistepScheduler"
-  ],
-  "text_encoder": [
-    "transformers",
-    "CLIPTextModel"
-  ],
-  "tokenizer": [
-    "transformers",
-    "CLIPTokenizer"
-  ],
-  "unet": [
-    "diffusers",
-    "UNet2DConditionModel"
-  ],
-  "vae": [
-    "diffusers",
-    "AutoencoderKL"
-  ]
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/blobs/2c2130b544c0c5a72d5d00da071ba130a9800fb2

@@ -1,24 +0,0 @@
-{
-  "bos_token": {
-    "content": "<|startoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "eos_token": {
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "pad_token": "<|endoftext|>",
-  "unk_token": {
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  }
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/blobs/2e8612a429492973fe60635b3f44a28b065cfac0

@@ -1,33 +0,0 @@
-{
-  "add_prefix_space": false,
-  "bos_token": {
-    "__type": "AddedToken",
-    "content": "<|startoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "clean_up_tokenization_spaces": true,
-  "do_lower_case": true,
-  "eos_token": {
-    "__type": "AddedToken",
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "errors": "replace",
-  "model_max_length": 77,
-  "pad_token": "<|endoftext|>",
-  "tokenizer_class": "CLIPTokenizer",
-  "unk_token": {
-    "__type": "AddedToken",
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  }
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/blobs/6d0f6f2e3f7f0133e7684e8dffd7d0dfb1f2b09ddaf327763214449cc7a36574

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:6d0f6f2e3f7f0133e7684e8dffd7d0dfb1f2b09ddaf327763214449cc7a36574
-size 1719125304

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/blobs/79d051410ce59a337cb358d79a3e81b9e0c691c34fcc7625ecfebf6a5db233a4

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:79d051410ce59a337cb358d79a3e81b9e0c691c34fcc7625ecfebf6a5db233a4
-size 167335342

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/blobs/7a834339c76ea80cd0d99560ed9cd188a1ac689c

@@ -1,32 +0,0 @@
-{
-  "_class_name": "AutoencoderKL",
-  "_diffusers_version": "0.21.0.dev0",
-  "_name_or_path": "/home/patrick/.cache/huggingface/hub/models--lykon-models--dreamshaper-7/snapshots/c4c9f9bec821e1862a78cbf45685cfb35b93638d/vae",
-  "act_fn": "silu",
-  "block_out_channels": [
-    128,
-    256,
-    512,
-    512
-  ],
-  "down_block_types": [
-    "DownEncoderBlock2D",
-    "DownEncoderBlock2D",
-    "DownEncoderBlock2D",
-    "DownEncoderBlock2D"
-  ],
-  "force_upcast": true,
-  "in_channels": 3,
-  "latent_channels": 4,
-  "layers_per_block": 2,
-  "norm_num_groups": 32,
-  "out_channels": 3,
-  "sample_size": 512,
-  "scaling_factor": 0.18215,
-  "up_block_types": [
-    "UpDecoderBlock2D",
-    "UpDecoderBlock2D",
-    "UpDecoderBlock2D",
-    "UpDecoderBlock2D"
-  ]
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/blobs/a63f334bd6446e1cd3d49f32dac82fd45d0fff53

@@ -1,23 +0,0 @@
-{
-  "_class_name": "DEISMultistepScheduler",
-  "_diffusers_version": "0.21.0.dev0",
-  "algorithm_type": "deis",
-  "beta_end": 0.012,
-  "beta_schedule": "scaled_linear",
-  "beta_start": 0.00085,
-  "clip_sample": false,
-  "dynamic_thresholding_ratio": 0.995,
-  "lower_order_final": true,
-  "num_train_timesteps": 1000,
-  "prediction_type": "epsilon",
-  "sample_max_value": 1.0,
-  "set_alpha_to_one": false,
-  "skip_prk_steps": true,
-  "solver_order": 2,
-  "solver_type": "logrho",
-  "steps_offset": 1,
-  "thresholding": false,
-  "timestep_spacing": "leading",
-  "trained_betas": null,
-  "use_karras_sigmas": false
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/blobs/a6f6744cfbcfe4fa9d236a231fd67e248389df7187dc15d52f16d9e9872105ff

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a6f6744cfbcfe4fa9d236a231fd67e248389df7187dc15d52f16d9e9872105ff
-size 246144152

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/blobs/bde6e9d55cd7617f5a97848de6cfd223e2f5a149

@@ -1,25 +0,0 @@
-{
-  "_name_or_path": "/home/patrick/.cache/huggingface/hub/models--lykon-models--dreamshaper-7/snapshots/c4c9f9bec821e1862a78cbf45685cfb35b93638d/text_encoder",
-  "architectures": [
-    "CLIPTextModel"
-  ],
-  "attention_dropout": 0.0,
-  "bos_token_id": 0,
-  "dropout": 0.0,
-  "eos_token_id": 2,
-  "hidden_act": "quick_gelu",
-  "hidden_size": 768,
-  "initializer_factor": 1.0,
-  "initializer_range": 0.02,
-  "intermediate_size": 3072,
-  "layer_norm_eps": 1e-05,
-  "max_position_embeddings": 77,
-  "model_type": "clip_text_model",
-  "num_attention_heads": 12,
-  "num_hidden_layers": 12,
-  "pad_token_id": 1,
-  "projection_dim": 768,
-  "torch_dtype": "float16",
-  "transformers_version": "4.33.0.dev0",
-  "vocab_size": 49408
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/blobs/ca62590a70013777bab17c188b5a96e6250b3002

@@ -1,66 +0,0 @@
-{
-  "_class_name": "UNet2DConditionModel",
-  "_diffusers_version": "0.21.0.dev0",
-  "_name_or_path": "/home/patrick/.cache/huggingface/hub/models--lykon-models--dreamshaper-7/snapshots/c4c9f9bec821e1862a78cbf45685cfb35b93638d/unet",
-  "act_fn": "silu",
-  "addition_embed_type": null,
-  "addition_embed_type_num_heads": 64,
-  "addition_time_embed_dim": null,
-  "attention_head_dim": 8,
-  "attention_type": "default",
-  "block_out_channels": [
-    320,
-    640,
-    1280,
-    1280
-  ],
-  "center_input_sample": false,
-  "class_embed_type": null,
-  "class_embeddings_concat": false,
-  "conv_in_kernel": 3,
-  "conv_out_kernel": 3,
-  "cross_attention_dim": 768,
-  "cross_attention_norm": null,
-  "down_block_types": [
-    "CrossAttnDownBlock2D",
-    "CrossAttnDownBlock2D",
-    "CrossAttnDownBlock2D",
-    "DownBlock2D"
-  ],
-  "downsample_padding": 1,
-  "dual_cross_attention": false,
-  "encoder_hid_dim": null,
-  "encoder_hid_dim_type": null,
-  "flip_sin_to_cos": true,
-  "freq_shift": 0,
-  "in_channels": 4,
-  "layers_per_block": 2,
-  "mid_block_only_cross_attention": null,
-  "mid_block_scale_factor": 1,
-  "mid_block_type": "UNetMidBlock2DCrossAttn",
-  "norm_eps": 1e-05,
-  "norm_num_groups": 32,
-  "num_attention_heads": null,
-  "num_class_embeds": null,
-  "only_cross_attention": false,
-  "out_channels": 4,
-  "projection_class_embeddings_input_dim": null,
-  "resnet_out_scale_factor": 1.0,
-  "resnet_skip_time_act": false,
-  "resnet_time_scale_shift": "default",
-  "sample_size": 64,
-  "time_cond_proj_dim": null,
-  "time_embedding_act_fn": null,
-  "time_embedding_dim": null,
-  "time_embedding_type": "positional",
-  "timestep_post_act": null,
-  "transformer_layers_per_block": 1,
-  "up_block_types": [
-    "UpBlock2D",
-    "CrossAttnUpBlock2D",
-    "CrossAttnUpBlock2D",
-    "CrossAttnUpBlock2D"
-  ],
-  "upcast_attention": null,
-  "use_linear_projection": false
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/blobs/e0675735fd0052745f2f7b7291e4aa7859389998

@@ -1,28 +0,0 @@
-{
-  "crop_size": {
-    "height": 224,
-    "width": 224
-  },
-  "do_center_crop": true,
-  "do_convert_rgb": true,
-  "do_normalize": true,
-  "do_rescale": true,
-  "do_resize": true,
-  "feature_extractor_type": "CLIPFeatureExtractor",
-  "image_mean": [
-    0.48145466,
-    0.4578275,
-    0.40821073
-  ],
-  "image_processor_type": "CLIPFeatureExtractor",
-  "image_std": [
-    0.26862954,
-    0.26130258,
-    0.27577711
-  ],
-  "resample": 3,
-  "rescale_factor": 0.00392156862745098,
-  "size": {
-    "shortest_edge": 224
-  }
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/refs/main

@@ -1 +0,0 @@
-9b481047f77996efa025e75e03941dbf51f506ad

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/snapshots/9b481047f77996efa025e75e03941dbf51f506ad/feature_extractor/preprocessor_config.json

@@ -1,28 +0,0 @@
-{
-  "crop_size": {
-    "height": 224,
-    "width": 224
-  },
-  "do_center_crop": true,
-  "do_convert_rgb": true,
-  "do_normalize": true,
-  "do_rescale": true,
-  "do_resize": true,
-  "feature_extractor_type": "CLIPFeatureExtractor",
-  "image_mean": [
-    0.48145466,
-    0.4578275,
-    0.40821073
-  ],
-  "image_processor_type": "CLIPFeatureExtractor",
-  "image_std": [
-    0.26862954,
-    0.26130258,
-    0.27577711
-  ],
-  "resample": 3,
-  "rescale_factor": 0.00392156862745098,
-  "size": {
-    "shortest_edge": 224
-  }
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/snapshots/9b481047f77996efa025e75e03941dbf51f506ad/model_index.json

@@ -1,34 +0,0 @@
-{
-  "_class_name": "StableDiffusionPipeline",
-  "_diffusers_version": "0.21.0.dev0",
-  "_name_or_path": "lykon-models/dreamshaper-7",
-  "feature_extractor": [
-    "transformers",
-    "CLIPFeatureExtractor"
-  ],
-  "requires_safety_checker": true,
-  "safety_checker": [
-    "stable_diffusion",
-    "StableDiffusionSafetyChecker"
-  ],
-  "scheduler": [
-    "diffusers",
-    "DEISMultistepScheduler"
-  ],
-  "text_encoder": [
-    "transformers",
-    "CLIPTextModel"
-  ],
-  "tokenizer": [
-    "transformers",
-    "CLIPTokenizer"
-  ],
-  "unet": [
-    "diffusers",
-    "UNet2DConditionModel"
-  ],
-  "vae": [
-    "diffusers",
-    "AutoencoderKL"
-  ]
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/snapshots/9b481047f77996efa025e75e03941dbf51f506ad/scheduler/scheduler_config.json

@@ -1,23 +0,0 @@
-{
-  "_class_name": "DEISMultistepScheduler",
-  "_diffusers_version": "0.21.0.dev0",
-  "algorithm_type": "deis",
-  "beta_end": 0.012,
-  "beta_schedule": "scaled_linear",
-  "beta_start": 0.00085,
-  "clip_sample": false,
-  "dynamic_thresholding_ratio": 0.995,
-  "lower_order_final": true,
-  "num_train_timesteps": 1000,
-  "prediction_type": "epsilon",
-  "sample_max_value": 1.0,
-  "set_alpha_to_one": false,
-  "skip_prk_steps": true,
-  "solver_order": 2,
-  "solver_type": "logrho",
-  "steps_offset": 1,
-  "thresholding": false,
-  "timestep_spacing": "leading",
-  "trained_betas": null,
-  "use_karras_sigmas": false
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/snapshots/9b481047f77996efa025e75e03941dbf51f506ad/text_encoder/config.json

@@ -1,25 +0,0 @@
-{
-  "_name_or_path": "/home/patrick/.cache/huggingface/hub/models--lykon-models--dreamshaper-7/snapshots/c4c9f9bec821e1862a78cbf45685cfb35b93638d/text_encoder",
-  "architectures": [
-    "CLIPTextModel"
-  ],
-  "attention_dropout": 0.0,
-  "bos_token_id": 0,
-  "dropout": 0.0,
-  "eos_token_id": 2,
-  "hidden_act": "quick_gelu",
-  "hidden_size": 768,
-  "initializer_factor": 1.0,
-  "initializer_range": 0.02,
-  "intermediate_size": 3072,
-  "layer_norm_eps": 1e-05,
-  "max_position_embeddings": 77,
-  "model_type": "clip_text_model",
-  "num_attention_heads": 12,
-  "num_hidden_layers": 12,
-  "pad_token_id": 1,
-  "projection_dim": 768,
-  "torch_dtype": "float16",
-  "transformers_version": "4.33.0.dev0",
-  "vocab_size": 49408
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/snapshots/9b481047f77996efa025e75e03941dbf51f506ad/text_encoder/model.fp16.safetensors

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a6f6744cfbcfe4fa9d236a231fd67e248389df7187dc15d52f16d9e9872105ff
-size 246144152

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/snapshots/9b481047f77996efa025e75e03941dbf51f506ad/tokenizer/special_tokens_map.json

@@ -1,24 +0,0 @@
-{
-  "bos_token": {
-    "content": "<|startoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "eos_token": {
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "pad_token": "<|endoftext|>",
-  "unk_token": {
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  }
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/snapshots/9b481047f77996efa025e75e03941dbf51f506ad/tokenizer/tokenizer_config.json

@@ -1,33 +0,0 @@
-{
-  "add_prefix_space": false,
-  "bos_token": {
-    "__type": "AddedToken",
-    "content": "<|startoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "clean_up_tokenization_spaces": true,
-  "do_lower_case": true,
-  "eos_token": {
-    "__type": "AddedToken",
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "errors": "replace",
-  "model_max_length": 77,
-  "pad_token": "<|endoftext|>",
-  "tokenizer_class": "CLIPTokenizer",
-  "unk_token": {
-    "__type": "AddedToken",
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  }
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/snapshots/9b481047f77996efa025e75e03941dbf51f506ad/unet/config.json

@@ -1,66 +0,0 @@
-{
-  "_class_name": "UNet2DConditionModel",
-  "_diffusers_version": "0.21.0.dev0",
-  "_name_or_path": "/home/patrick/.cache/huggingface/hub/models--lykon-models--dreamshaper-7/snapshots/c4c9f9bec821e1862a78cbf45685cfb35b93638d/unet",
-  "act_fn": "silu",
-  "addition_embed_type": null,
-  "addition_embed_type_num_heads": 64,
-  "addition_time_embed_dim": null,
-  "attention_head_dim": 8,
-  "attention_type": "default",
-  "block_out_channels": [
-    320,
-    640,
-    1280,
-    1280
-  ],
-  "center_input_sample": false,
-  "class_embed_type": null,
-  "class_embeddings_concat": false,
-  "conv_in_kernel": 3,
-  "conv_out_kernel": 3,
-  "cross_attention_dim": 768,
-  "cross_attention_norm": null,
-  "down_block_types": [
-    "CrossAttnDownBlock2D",
-    "CrossAttnDownBlock2D",
-    "CrossAttnDownBlock2D",
-    "DownBlock2D"
-  ],
-  "downsample_padding": 1,
-  "dual_cross_attention": false,
-  "encoder_hid_dim": null,
-  "encoder_hid_dim_type": null,
-  "flip_sin_to_cos": true,
-  "freq_shift": 0,
-  "in_channels": 4,
-  "layers_per_block": 2,
-  "mid_block_only_cross_attention": null,
-  "mid_block_scale_factor": 1,
-  "mid_block_type": "UNetMidBlock2DCrossAttn",
-  "norm_eps": 1e-05,
-  "norm_num_groups": 32,
-  "num_attention_heads": null,
-  "num_class_embeds": null,
-  "only_cross_attention": false,
-  "out_channels": 4,
-  "projection_class_embeddings_input_dim": null,
-  "resnet_out_scale_factor": 1.0,
-  "resnet_skip_time_act": false,
-  "resnet_time_scale_shift": "default",
-  "sample_size": 64,
-  "time_cond_proj_dim": null,
-  "time_embedding_act_fn": null,
-  "time_embedding_dim": null,
-  "time_embedding_type": "positional",
-  "timestep_post_act": null,
-  "transformer_layers_per_block": 1,
-  "up_block_types": [
-    "UpBlock2D",
-    "CrossAttnUpBlock2D",
-    "CrossAttnUpBlock2D",
-    "CrossAttnUpBlock2D"
-  ],
-  "upcast_attention": null,
-  "use_linear_projection": false
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/snapshots/9b481047f77996efa025e75e03941dbf51f506ad/unet/diffusion_pytorch_model.fp16.safetensors

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:6d0f6f2e3f7f0133e7684e8dffd7d0dfb1f2b09ddaf327763214449cc7a36574
-size 1719125304

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/snapshots/9b481047f77996efa025e75e03941dbf51f506ad/vae/config.json

@@ -1,32 +0,0 @@
-{
-  "_class_name": "AutoencoderKL",
-  "_diffusers_version": "0.21.0.dev0",
-  "_name_or_path": "/home/patrick/.cache/huggingface/hub/models--lykon-models--dreamshaper-7/snapshots/c4c9f9bec821e1862a78cbf45685cfb35b93638d/vae",
-  "act_fn": "silu",
-  "block_out_channels": [
-    128,
-    256,
-    512,
-    512
-  ],
-  "down_block_types": [
-    "DownEncoderBlock2D",
-    "DownEncoderBlock2D",
-    "DownEncoderBlock2D",
-    "DownEncoderBlock2D"
-  ],
-  "force_upcast": true,
-  "in_channels": 3,
-  "latent_channels": 4,
-  "layers_per_block": 2,
-  "norm_num_groups": 32,
-  "out_channels": 3,
-  "sample_size": 512,
-  "scaling_factor": 0.18215,
-  "up_block_types": [
-    "UpDecoderBlock2D",
-    "UpDecoderBlock2D",
-    "UpDecoderBlock2D",
-    "UpDecoderBlock2D"
-  ]
-}

extensions/sd_webui_realtime_lcm_canvas/scripts/models/models--Lykon--dreamshaper-7/snapshots/9b481047f77996efa025e75e03941dbf51f506ad/vae/diffusion_pytorch_model.fp16.safetensors

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:79d051410ce59a337cb358d79a3e81b9e0c691c34fcc7625ecfebf6a5db233a4
-size 167335342

extensions/sd_webui_realtime_lcm_canvas/scripts/realtime_lcm_canvas.py

@@ -1,139 +0,0 @@
-import os
-import time
-import torch
-import random
-import gradio as gr
-
-from sys import platform
-from PIL import Image
-from contextlib import nullcontext
-from diffusers.utils import load_image
-from diffusers import AutoPipelineForImage2Image, LCMScheduler
-from modules import script_callbacks
-
-cache_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "models")
-
-os.environ["TRANSFORMERS_CACHE"] = cache_path
-os.environ["HF_HUB_CACHE"] = cache_path
-os.environ["HF_HOME"] = cache_path
-is_mac = platform == "darwin"
-
-
-def should_use_fp16():
-    if is_mac:
-        return True
-
-    gpu_props = torch.cuda.get_device_properties("cuda")
-    if gpu_props.major < 6:
-        return False
-
-    nvidia_16_series = ["1660", "1650", "1630"]
-    for x in nvidia_16_series:
-        if x in gpu_props.name:
-            return False
-
-    return True
-
-
-class timer:
-    def __init__(self, method_name="timed process"):
-        self.method = method_name
-
-    def __enter__(self):
-        self.start = time.time()
-        print(f"{self.method} starts")
-
-    def __exit__(self, exc_type, exc_val, exc_tb):
-        end = time.time()
-        print(f"{self.method} took {str(round(end - self.start, 2))}s")
-
-
-def load_models(model_id="Lykon/dreamshaper-7"):
-    if not is_mac:
-        torch.backends.cuda.matmul.allow_tf32 = True
-
-    use_fp16 = should_use_fp16()
-    lcm_lora_id = "latent-consistency/lcm-lora-sdv1-5"
-
-    if use_fp16:
-        pipe = AutoPipelineForImage2Image.from_pretrained(
-            model_id, cache_dir=cache_path, torch_dtype=torch.float16, variant="fp16", safety_checker=None)
-    else:
-        pipe = AutoPipelineForImage2Image.from_pretrained(
-            model_id, cache_dir=cache_path, safety_checker=None)
-
-    pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
-    pipe.load_lora_weights(lcm_lora_id)
-    pipe.fuse_lora()
-
-    device = "mps" if is_mac else "cuda"
-    pipe.to(device=device)
-    generator = torch.Generator()
-
-    def inference(prompt,
-                  img,
-                  steps=4,
-                  cfg=1,
-                  sketch_strength=0.9,
-                  seed=random.randrange(0, 2 ** 63)):
-        with torch.inference_mode():
-            with torch.autocast("cuda") if device == "cuda" else nullcontext():
-                with timer("inference"):
-                    return pipe(
-                        prompt=prompt,
-                        image=load_image(img),
-                        generator=generator.manual_seed(seed),
-                        num_inference_steps=steps,
-                        guidance_scale=cfg,
-                        strength=sketch_strength
-                    ).images[0]
-
-    return inference
-
-
-def on_ui_tabs():
-    canvas_size = 512
-    inference = load_models()
-
-    with gr.Blocks(analytics_enabled=False) as realtime_lcm_canvas:
-        with gr.Column():
-            with gr.Row():
-                with gr.Column():
-                    steps = gr.Slider(label="steps", minimum=4, maximum=8, step=1, value=4, interactive=True)
-                    cfg = gr.Slider(label="cfg", minimum=0.1, maximum=3, step=0.1, value=1, interactive=True)
-                    sketch_strength = gr.Slider(label="sketch strength", minimum=0.1, maximum=0.9, step=0.1, value=0.9,
-                                                interactive=True)
-                with gr.Column():
-                    model_id = gr.Text(
-                        label="Model Hugging Face id (after changing this wait until the model downloads in the console)",
-                        value="Lykon/dreamshaper-7", interactive=True)
-                    prompt = gr.Text(label="Prompt",
-                                     value="Scary warewolf, 8K, realistic, colorful, long sharp teeth, splash art",
-                                     interactive=True)
-                    seed = gr.Number(label="seed", value=1337, interactive=True)
-
-            with gr.Row(equal_height=True):
-                img = gr.Image(source="canvas", tool="color-sketch", shape=(canvas_size, canvas_size),
-                               width=canvas_size,
-                               height=canvas_size, type="pil")
-                output = gr.Image(width=canvas_size, height=canvas_size)
-
-                def process_image(prompt, img, steps, cfg, sketch_strength, seed):
-                    if not img:
-                        return Image.new("RGB", (canvas_size, canvas_size))
-                    return inference(prompt, img, steps, cfg, sketch_strength, seed=int(seed))
-
-                reactive_controls = [prompt, img, steps, cfg, sketch_strength, seed]
-                for control in reactive_controls:
-                    control.change(fn=process_image, inputs=reactive_controls, outputs=output)
-
-                def update_model(model_name):
-                    global inference
-                    inference = load_models(model_name)
-
-                model_id.change(fn=update_model, inputs=model_id)
-
-    return [(realtime_lcm_canvas, "Realtime_LCM_Canvas", "Realtime_LCM_Canvas")]
-
-
-script_callbacks.on_ui_tabs(on_ui_tabs)