DiZH797 commited on
Commit
f0cbac2
·
verified ·
1 Parent(s): 8d70b93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ from typing import Optional
14
  # кэш для пайплайнов (чтобы не перезагружать модель при каждом запросе)
15
  PIPE_CACHE: dict[str, DiffusionPipeline] = {}
16
  DEFAULT_MODEL = "CompVis/stable-diffusion-v1-4"
17
- BASE_MODEL_FOR_LORA = "CompVis/stable-diffusion-v1-5" # Base model used for LoRA training
18
  LORA_MODEL_ID = "DiZH797/my-tuned-lora" # Your uploaded LoRA model ID
19
  MODEL_OPTIONS = [
20
  "CompVis/stable-diffusion-v1-4",
 
14
  # кэш для пайплайнов (чтобы не перезагружать модель при каждом запросе)
15
  PIPE_CACHE: dict[str, DiffusionPipeline] = {}
16
  DEFAULT_MODEL = "CompVis/stable-diffusion-v1-4"
17
+ BASE_MODEL_FOR_LORA = "stable-diffusion-v1-5/stable-diffusion-v1-5" # Base model used for LoRA training
18
  LORA_MODEL_ID = "DiZH797/my-tuned-lora" # Your uploaded LoRA model ID
19
  MODEL_OPTIONS = [
20
  "CompVis/stable-diffusion-v1-4",