Update convert.py

remove useless comment
This commit is contained in:
Ivan Stepanov 2023-05-04 13:49:15 +03:00 committed by GitHub
parent dd8902d3e4
commit e3c2421b9f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -709,9 +709,7 @@ class LazyUnpickler(pickle.Unpickler):
@staticmethod
def rebuild_from_type_v2(func, new_type, args, state):
# import torch; return torch._tensor._rebuild_from_type_v2(func, new_type, args, state)
ret = func(*args)
return ret
return func(*args)
CLASSES: Dict[Any, Any] = {
('torch._tensor', '_rebuild_from_type_v2'): rebuild_from_type_v2,