yanze commited on
Commit
5b1962e
1 Parent(s): 2f5b111

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -283,8 +283,7 @@ if __name__ == "__main__":
283
  parser = argparse.ArgumentParser(description="PuLID for FLUX.1-dev")
284
  parser.add_argument("--name", type=str, default="flux-dev", choices=list('flux-dev'),
285
  help="currently only support flux-dev")
286
- parser.add_argument("--device", type=str, default="cuda" if torch.cuda.is_available() else "cpu",
287
- help="Device to use")
288
  parser.add_argument("--offload", action="store_true", help="Offload model to CPU when not in use")
289
  parser.add_argument("--port", type=int, default=8080, help="Port to use")
290
  parser.add_argument("--dev", action='store_true', help="Development mode")
 
283
  parser = argparse.ArgumentParser(description="PuLID for FLUX.1-dev")
284
  parser.add_argument("--name", type=str, default="flux-dev", choices=list('flux-dev'),
285
  help="currently only support flux-dev")
286
+ parser.add_argument("--device", type=str, default="cuda", help="Device to use")
 
287
  parser.add_argument("--offload", action="store_true", help="Offload model to CPU when not in use")
288
  parser.add_argument("--port", type=int, default=8080, help="Port to use")
289
  parser.add_argument("--dev", action='store_true', help="Development mode")