improve anime

Improve Fooocus Anime a bit by using better SD1.5 refining formulation.
This commit is contained in:
lllyasviel
2023-10-21 17:11:55 -07:00
parent 60c05342b2
commit 736a5aa3ac
6 changed files with 36 additions and 59 deletions
+3 -4
View File
@@ -69,7 +69,7 @@ vae_approx_filename = os.path.join(vae_approx_path, 'xl-to-v1_interposer-v3.1.sa
def parse(x):
global vae_approx_model
x_origin = x['samples'].clone()
x_origin = x.clone()
if vae_approx_model is None:
model = Interposer()
@@ -89,6 +89,5 @@ def parse(x):
fcbh.model_management.load_model_gpu(vae_approx_model)
x = x_origin.to(device=vae_approx_model.load_device, dtype=vae_approx_model.dtype)
x = vae_approx_model.model(x)
return {'samples': x.to(x_origin)}
x = vae_approx_model.model(x).to(x_origin)
return x