Hugging Face's logo

Spaces:
diffusers
/
unofficial-SDXL-Turbo-i2i-t2i
Running on A10G

App Files Community
26
phamalor commited on
Commit
cf8b7bf
·
verified ·
1 Parent(s): 45b7c7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -39,25 +39,25 @@ if SAFETY_CHECKER == "True":
39
  i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
40
  "stabilityai/sdxl-turbo",
41
  torch_dtype=torch_dtype,
42
- variant="fp16" if torch_dtype == torch.float16 else "fp32",
43
  )
44
  t2i_pipe = AutoPipelineForText2Image.from_pretrained(
45
  "stabilityai/sdxl-turbo",
46
  torch_dtype=torch_dtype,
47
- variant="fp16" if torch_dtype == torch.float16 else "fp32",
48
  )
49
  else:
50
  i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
51
  "stabilityai/sdxl-turbo",
52
  safety_checker=None,
53
  torch_dtype=torch_dtype,
54
- variant="fp16" if torch_dtype == torch.float16 else "fp32",
55
  )
56
  t2i_pipe = AutoPipelineForText2Image.from_pretrained(
57
  "stabilityai/sdxl-turbo",
58
  safety_checker=None,
59
  torch_dtype=torch_dtype,
60
- variant="fp16" if torch_dtype == torch.float16 else "fp32",
61
  )
62
 
63
 
 
39
  i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
40
  "stabilityai/sdxl-turbo",
41
  torch_dtype=torch_dtype,
42
+ variant="fp30" if torch_dtype == torch.float16 else "fp60",
43
  )
44
  t2i_pipe = AutoPipelineForText2Image.from_pretrained(
45
  "stabilityai/sdxl-turbo",
46
  torch_dtype=torch_dtype,
47
+ variant="fp30" if torch_dtype == torch.float16 else "fp60",
48
  )
49
  else:
50
  i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
51
  "stabilityai/sdxl-turbo",
52
  safety_checker=None,
53
  torch_dtype=torch_dtype,
54
+ variant="fp30" if torch_dtype == torch.float16 else "fp60",
55
  )
56
  t2i_pipe = AutoPipelineForText2Image.from_pretrained(
57
  "stabilityai/sdxl-turbo",
58
  safety_checker=None,
59
  torch_dtype=torch_dtype,
60
+ variant="fp30" if torch_dtype == torch.float16 else "fp60",
61
  )
62
 
63