lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-28 02:33:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

revise inference dtype

Browse Source
This commit is contained in:
layerdiffusion
2024-08-07 17:08:47 -07:00
parent 89ea50a19d
commit b61bf553ea
5 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
backend/diffusion_engine/flux.py
View File

@@ -38,7 +38,8 @@ class Flux(ForgeDiffusionEngine):
unet = UnetPatcher.from_model(
model=huggingface_components['transformer'],
diffusers_scheduler=None,
k_predictor=PredictionFlux(sigma_data=1.0, prediction_type='const', shift=1.15, timesteps=10000)
k_predictor=PredictionFlux(sigma_data=1.0, prediction_type='const', shift=1.15, timesteps=10000),
config=estimated_config
)
self.text_processing_engine_l = ClassicTextProcessingEngine(

3
backend/diffusion_engine/sd15.py
View File

@@ -29,7 +29,8 @@ class StableDiffusion(ForgeDiffusionEngine):
unet = UnetPatcher.from_model(
model=huggingface_components['unet'],
diffusers_scheduler=huggingface_components['scheduler']
diffusers_scheduler=huggingface_components['scheduler'],
config=estimated_config
)
self.text_processing_engine = ClassicTextProcessingEngine(

3
backend/diffusion_engine/sd20.py
View File

@@ -29,7 +29,8 @@ class StableDiffusion2(ForgeDiffusionEngine):
unet = UnetPatcher.from_model(
model=huggingface_components['unet'],
diffusers_scheduler=huggingface_components['scheduler']
diffusers_scheduler=huggingface_components['scheduler'],
config=estimated_config
)
self.text_processing_engine = ClassicTextProcessingEngine(

3
backend/diffusion_engine/sdxl.py
View File

@@ -32,7 +32,8 @@ class StableDiffusionXL(ForgeDiffusionEngine):
unet = UnetPatcher.from_model(
model=huggingface_components['unet'],
diffusers_scheduler=huggingface_components['scheduler']
diffusers_scheduler=huggingface_components['scheduler'],
config=estimated_config
)
self.text_processing_engine_l = ClassicTextProcessingEngine(