Spaces:

smartfeed
/

turbo_inversion

Sleeping

zhiweili commited on Sep 22

Commit

ad5e86e

•

1 Parent(s): d7867bc

change base model

Files changed (1) hide show

inversion_run_realvxl.py CHANGED Viewed

@@ -48,7 +48,7 @@ assert (
 generator = None
 device = "cuda" if torch.cuda.is_available() else "cpu"
-BASE_MODEL = "SG161222/RealVisXL_V5.0_Lightning"
 pipeline = StableDiffusionXLImg2ImgPipeline.from_pretrained(
     BASE_MODEL,

 generator = None
 device = "cuda" if torch.cuda.is_available() else "cpu"
+BASE_MODEL = "stabilityai/sdxl-turbo"
 pipeline = StableDiffusionXLImg2ImgPipeline.from_pretrained(
     BASE_MODEL,