Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -157,7 +157,10 @@ def _gen_shape(
|
|
| 157 |
save_folder = gen_save_folder()
|
| 158 |
image = rmbg_worker(rgb_image)
|
| 159 |
|
| 160 |
-
|
|
|
|
|
|
|
|
|
|
| 161 |
|
| 162 |
outputs = i23d_worker(
|
| 163 |
image=image,
|
|
|
|
| 157 |
save_folder = gen_save_folder()
|
| 158 |
image = rmbg_worker(rgb_image)
|
| 159 |
|
| 160 |
+
if args.device in ["cuda", "cpu"]:
|
| 161 |
+
generator = torch.Generator(device=args.device).manual_seed(seed)
|
| 162 |
+
else:
|
| 163 |
+
generator = torch.Generator().manual_seed(seed) # fallback für MPS
|
| 164 |
|
| 165 |
outputs = i23d_worker(
|
| 166 |
image=image,
|