Spaces:
Runtime error
Runtime error
Commit
·
6296a21
1
Parent(s):
6bae932
structure
Browse files
config.py
CHANGED
@@ -5,8 +5,8 @@ from pydantic import BaseModel
|
|
5 |
|
6 |
class Settings(BaseModel):
|
7 |
hf_token: str = os.environ.get("hf_token")
|
8 |
-
base_sd_model: str = os.environ.get("base_sd_model"
|
9 |
-
refiner_sd_model: str = os.environ.get("refiner_sd_model"
|
10 |
version: str = "0.1.0"
|
11 |
url_version: str = "v1"
|
12 |
prefix: str = "v1/unik-ml"
|
|
|
5 |
|
6 |
class Settings(BaseModel):
|
7 |
hf_token: str = os.environ.get("hf_token")
|
8 |
+
base_sd_model: str = os.environ.get("base_sd_model") or "stabilityai/stable-diffusion-xl-base-1.0"
|
9 |
+
refiner_sd_model: str = os.environ.get("refiner_sd_model") or "stabilityai/stable-diffusion-xl-refiner-1.0"
|
10 |
version: str = "0.1.0"
|
11 |
url_version: str = "v1"
|
12 |
prefix: str = "v1/unik-ml"
|