Spaces:
Sleeping
Sleeping
Update gradio_sd3.py
Browse files- 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.")
|