Commit 2f5f98bb authored by zhuwenwen's avatar zhuwenwen
Browse files

remove torch version info

parent 7a8923fb
...@@ -36,7 +36,6 @@ def get_model_architecture( ...@@ -36,7 +36,6 @@ def get_model_architecture(
else: else:
os.environ['AWQ_PAD'] = '1' os.environ['AWQ_PAD'] = '1'
except Exception as e: except Exception as e:
print("Info: this version torch cannot get eco device info.\n")
if os.getenv('AWQ_PAD') != '0': if os.getenv('AWQ_PAD') != '0':
os.environ['AWQ_PAD'] = '1' os.environ['AWQ_PAD'] = '1'
else: else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment