LejobuildYT commited on
Commit
014ac33
·
verified ·
1 Parent(s): f83328a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -157,7 +157,10 @@ def _gen_shape(
157
  save_folder = gen_save_folder()
158
  image = rmbg_worker(rgb_image)
159
 
160
- generator = torch.Generator(device=args.device).manual_seed(seed)
 
 
 
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,