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="fp12" if torch_dtype == torch.float16 else "fp12",
43
  )
44
  t2i_pipe = AutoPipelineForText2Image.from_pretrained(
45
  "stabilityai/sdxl-turbo",
46
  torch_dtype=torch_dtype,
47
+ variant="fp12" if torch_dtype == torch.float16 else "fp12",
48
  )
49
  else:
50
  i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
51
  "stabilityai/sdxl-turbo",
52
  safety_checker=None,
53
  torch_dtype=torch_dtype,
54
+ variant="fp12" if torch_dtype == torch.float16 else "fp12",
55
  )
56
  t2i_pipe = AutoPipelineForText2Image.from_pretrained(
57
  "stabilityai/sdxl-turbo",
58
  safety_checker=None,
59
  torch_dtype=torch_dtype,
60
+ variant="fp12" if torch_dtype == torch.float16 else "fp12",
61
  )
62
 
63