skygpu
/
skynet
mirror of https://github.com/skygpu/skynet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add param for stablexl

pull/11/head
Konstantine Tsafatinos 2023-07-19 16:14:03 -04:00
parent ae348c7c6f
commit 08da0681cd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
skynet/utils.py
View File

@ -79,7 +79,7 @@ def pipeline_for(model: str, mem_fraction: float = 1.0, image=False) -> Diffusio
if model == 'snowkidy/stable-diffusion-xl-base-0.9': if model == 'snowkidy/stable-diffusion-xl-base-0.9':
# TODO: figure out what this does # TODO: figure out what this does
params['addition_embed_type'] = None params['addition_embed_type'] = { 'text_time': None }
if model == 'runwayml/stable-diffusion-v1-5': if model == 'runwayml/stable-diffusion-v1-5':
params['revision'] = 'fp16' params['revision'] = 'fp16'