Update main.py

This commit is contained in:
Paillat
2023-02-01 09:20:20 +01:00
parent af984608d7
commit 4fddc74842

View File

@@ -14,7 +14,7 @@ path=""
prompt="" prompt=""
app = FastAPI() app = FastAPI()
context = sdkit.Context() context = sdkit.Context()
context.model_paths['stable-diffusion'] = '.\\models\\stable-diffusion\\openjourney-v2-unpruned.ckpt' context.model_paths['stable-diffusion'] = '.\\models\\stable-diffusion\\sd-v1-4.ckpt'
load_model(context, 'stable-diffusion') load_model(context, 'stable-diffusion')
context.model_paths['gfpgan'] = '.\\models\\gfpgan\\GFPGANv1.3.pth' context.model_paths['gfpgan'] = '.\\models\\gfpgan\\GFPGANv1.3.pth'
load_model(context, 'gfpgan') load_model(context, 'gfpgan')