Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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"
|
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")
|