Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|