Clone04 commited on
Commit
8a77b59
·
verified ·
1 Parent(s): e56840f

Update gradio_sd3.py

Browse files
Files changed (1) hide show
  1. gradio_sd3.py +1 -1
gradio_sd3.py CHANGED
@@ -305,7 +305,7 @@ def create_demo(model_path, device, offload, aggressive_offload, with_fp16):
305
  if __name__ == "__main__":
306
  import argparse
307
  parser = argparse.ArgumentParser(description="FitDiT")
308
- parser.add_argument("--model_path", type=str, required=True, help="The path of FitDiT model.")
309
  parser.add_argument("--device", type=str, default="cuda:0", help="Device to use")
310
  parser.add_argument("--fp16", action="store_true", help="Load model with fp16, default is bf16")
311
  parser.add_argument("--offload", action="store_true", help="Offload model to CPU when not in use.")
 
305
  if __name__ == "__main__":
306
  import argparse
307
  parser = argparse.ArgumentParser(description="FitDiT")
308
+ parser.add_argument("--model_path", type=str, default="BoyuanJiang/FitDiT", required=True, help="The path of FitDiT model.")
309
  parser.add_argument("--device", type=str, default="cuda:0", help="Device to use")
310
  parser.add_argument("--fp16", action="store_true", help="Load model with fp16, default is bf16")
311
  parser.add_argument("--offload", action="store_true", help="Offload model to CPU when not in use.")