Spaces:

KingNish
/

SDXL-Flash

Running on Zero

KingNish commited on May 22

Commit

f87fd5b

•

1 Parent(s): e58f937

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,8 +29,7 @@ if torch.cuda.is_available():
         "sd-community/sdxl-flash",
         torch_dtype=torch.float16,
         use_safetensors=True,
-        add_watermarker=False,
-        variant="fp16"
     )
     if ENABLE_CPU_OFFLOAD:
         pipe.enable_model_cpu_offload()

         "sd-community/sdxl-flash",
         torch_dtype=torch.float16,
         use_safetensors=True,
+        add_watermarker=False
     )
     if ENABLE_CPU_OFFLOAD:
         pipe.enable_model_cpu_offload()