-
Notifications
You must be signed in to change notification settings - Fork 4.3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[BUG]: TypeError: _gen_python_code() got an unexpected keyword argument 'verbose' #5673
Labels
bug
Something isn't working
Comments
Thanks for the answer, I pulled colossalai==0.3.7, when torch==2.2.1,The following error occurs
when the torch==2.1.1,The following error occurs
|
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
🐛 Describe the bug
When I run examples/language/gpt/experiments/auto_parallel/auto_parallel_with_gpt.py,The error is as follows
File "/opt/conda/lib/python3.9/site-packages/colossalai/auto_parallel/tensor_shard/initialize.py", line 265, in initialize_model
gm = ColoGraphModule(model, graph, model.class.name)
File "/opt/conda/lib/python3.9/site-packages/colossalai/_analyzer/fx/graph_module.py", line 110, in init
super().init(root, graph, class_name)
File "/opt/conda/lib/python3.9/site-packages/torch/fx/graph_module.py", line 385, in init
self.graph = graph
File "/opt/conda/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1754, in setattr
super().setattr(name, value)
File "/opt/conda/lib/python3.9/site-packages/torch/fx/graph_module.py", line 426, in graph
self.recompile()
File "/opt/conda/lib/python3.9/site-packages/colossalai/_analyzer/fx/graph_module.py", line 141, in recompile
python_code = self._graph.python_code(root_module="self",verbose=True)
File "/opt/conda/lib/python3.9/site-packages/torch/fx/graph.py", line 1268, in python_code
return self._python_code(root_module, namespace, verbose=verbose)
File "/opt/conda/lib/python3.9/site-packages/torch/fx/graph.py", line 1271, in _python_code
return self._codegen._gen_python_code(self.nodes, root_module, namespace, verbose=verbose)
TypeError: _gen_python_code() got an unexpected keyword argument 'verbose'
Environment
torch==2.1.0
colossalai==0.3.6
The text was updated successfully, but these errors were encountered: