1inkusFace commited on
Commit
ffc38e3
·
verified ·
1 Parent(s): 6a08ac3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -115,7 +115,7 @@ pipe = StableDiffusion3Pipeline.from_single_file(
115
  #pipe.to(device=device) #, dtype=torch.bfloat16)
116
  #pipe.vae=vaeX #.to('cpu')
117
 
118
- pipe.to(device)
119
  upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cuda'))
120
 
121
  MAX_SEED = np.iinfo(np.int32).max
 
115
  #pipe.to(device=device) #, dtype=torch.bfloat16)
116
  #pipe.vae=vaeX #.to('cpu')
117
 
118
+ pipe.to(device,torch.bfloat16)
119
  upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cuda'))
120
 
121
  MAX_SEED = np.iinfo(np.int32).max