Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ from typing import Optional
|
|
| 11 |
PIPE_CACHE: dict[str, DiffusionPipeline] = {}
|
| 12 |
DEFAULT_MODEL = "CompVis/stable-diffusion-v1-4"
|
| 13 |
BASE_MODEL_FOR_LORA = "CompVis/stable-diffusion-v1-4" # Base model used for LoRA training
|
| 14 |
-
LORA_MODEL_ID = "DiZH797/
|
| 15 |
MODEL_OPTIONS = [
|
| 16 |
"CompVis/stable-diffusion-v1-4",
|
| 17 |
"stabilityai/stable-diffusion-2-1",
|
|
|
|
| 11 |
PIPE_CACHE: dict[str, DiffusionPipeline] = {}
|
| 12 |
DEFAULT_MODEL = "CompVis/stable-diffusion-v1-4"
|
| 13 |
BASE_MODEL_FOR_LORA = "CompVis/stable-diffusion-v1-4" # Base model used for LoRA training
|
| 14 |
+
LORA_MODEL_ID = "DiZH797/my-tuned-lora" # Your uploaded LoRA model ID
|
| 15 |
MODEL_OPTIONS = [
|
| 16 |
"CompVis/stable-diffusion-v1-4",
|
| 17 |
"stabilityai/stable-diffusion-2-1",
|