Fabrice-TIERCELIN commited on
Commit
7d92b01
1 Parent(s): 36a8419

Fabrice-TIERCELIN/sdxl-turbo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
15
  floatType = torch.float16 if torch.cuda.is_available() else torch.float32
16
  variant = "fp16" if torch.cuda.is_available() else None
17
 
18
- pipe = AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype = floatType, variant = variant)
19
  pipe = pipe.to(device)
20
 
21
  def update_seed(is_randomize_seed, seed):
 
15
  floatType = torch.float16 if torch.cuda.is_available() else torch.float32
16
  variant = "fp16" if torch.cuda.is_available() else None
17
 
18
+ pipe = AutoPipelineForImage2Image.from_pretrained("Fabrice-TIERCELIN/sdxl-turbo", torch_dtype = floatType, variant = variant)
19
  pipe = pipe.to(device)
20
 
21
  def update_seed(is_randomize_seed, seed):