extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-01-23 22:58:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

Formatting changes.

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-22 19:02:02 -04:00
parent 026dd38480
commit 8adf965d0b
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
ui/sd_internal/runtime.py
View File

@ -171,10 +171,10 @@ def load_model_ckpt():
if not thread_data.precision: if not thread_data.precision:
thread_data.precision = 'full' if thread_data.force_full_precision else 'autocast' thread_data.precision = 'full' if thread_data.force_full_precision else 'autocast'
if not thread_data.unet_bs: if not thread_data.unet_bs:
thread_data.unet_bs = 1 thread_data.unet_bs = 1
if thread_data.device == 'cpu': if thread_data.device == 'cpu':
thread_data.precision = 'full' thread_data.precision = 'full'
@ -207,9 +207,9 @@ def load_model_ckpt():
model.turbo = thread_data.turbo model.turbo = thread_data.turbo
if thread_data.device != 'cpu': if thread_data.device != 'cpu':
model.to(thread_data.device) model.to(thread_data.device)
#if thread_data.reduced_memory: #if thread_data.reduced_memory:
#model.model1.to("cpu") #model.model1.to("cpu")
#model.model2.to("cpu") #model.model2.to("cpu")
thread_data.model = model thread_data.model = model
modelCS = instantiate_from_config(config.modelCondStage) modelCS = instantiate_from_config(config.modelCondStage)
@ -464,7 +464,7 @@ def do_mk_img(req: Request):
thread_data.turbo = req.turbo thread_data.turbo = req.turbo
thread_data.model.turbo = req.turbo thread_data.model.turbo = req.turbo
# Start by cleaning memory, loading and unloading things can leave memory allocated. # Start by cleaning memory, loading and unloading things can leave memory allocated.
gc() gc()
opt_prompt = req.prompt opt_prompt = req.prompt