Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = "
|
| 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",
|