prithivMLmods commited on
Commit
a8f7729
1 Parent(s): 103ac02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ ENABLE_CPU_OFFLOAD = 0
46
  if torch.cuda.is_available():
47
  pipe = StableDiffusionXLPipeline.from_pretrained(
48
  "SG161222/RealVisXL_V5.0_Lightning",
49
- torch_dtype=torch.float32,
50
  use_safetensors=True,
51
  )
52
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
46
  if torch.cuda.is_available():
47
  pipe = StableDiffusionXLPipeline.from_pretrained(
48
  "SG161222/RealVisXL_V5.0_Lightning",
49
+ torch_dtype=torch.float16,
50
  use_safetensors=True,
51
  )
52
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)