mirror of
https://github.com/bigscience-workshop/petals
synced 2024-11-11 19:11:04 +00:00
add identity path
This commit is contained in:
parent
35310698f0
commit
57f4e0a899
@ -49,6 +49,7 @@ def main():
|
|||||||
|
|
||||||
parser.add_argument('--custom_module_path', type=str, required=False,
|
parser.add_argument('--custom_module_path', type=str, required=False,
|
||||||
help='Path of a file with custom nn.modules, wrapped into special decorator')
|
help='Path of a file with custom nn.modules, wrapped into special decorator')
|
||||||
|
parser.add_argument('--identity_path', type=str, required=False, help='Path to identity file to be used in P2P')
|
||||||
|
|
||||||
# fmt:on
|
# fmt:on
|
||||||
args = vars(parser.parse_args())
|
args = vars(parser.parse_args())
|
||||||
|
Loading…
Reference in New Issue
Block a user