RoniFinTech commited on
Commit
6296a21
·
1 Parent(s): 6bae932
Files changed (1) hide show
  1. config.py +2 -2
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", "stabilityai/stable-diffusion-xl-base-1.0")
9
- refiner_sd_model: str = os.environ.get("refiner_sd_model", "stabilityai/stable-diffusion-xl-refiner-1.0")
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"