Spaces:

KingNish
/

SDXL-Flash

Running on Zero

KingNish commited on Jun 3

Commit

77f1348

•

1 Parent(s): 085c378

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,6 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
     torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
     use_safetensors=True,
     add_watermarker=False,
-    variant="fp16" if torch.cuda.is_available() else None,
 ).to(device)
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)

     torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
     use_safetensors=True,
     add_watermarker=False,
 ).to(device)
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)