huckiyang commited on
Commit
b76263f
·
1 Parent(s): 05d3571

transformer version check

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -18,6 +18,9 @@ import sys
18
  login(token=os.environ.get("LA_NAME"))
19
  os.environ["LASER"] = "laser"
20
 
 
 
 
21
  def check_and_install(package, required_version):
22
  try:
23
  dist = pkg_resources.get_distribution(package)
@@ -32,7 +35,9 @@ def check_and_install(package, required_version):
32
  subprocess.check_call([sys.executable, "-m", "pip", "install", f"{package}=={required_version}"])
33
  packages = {
34
  "pip": "24.0",
35
- "fairseq": "0.12.2"
 
 
36
  }
37
  for package, version in packages.items():
38
  check_and_install(package, version)
 
18
  login(token=os.environ.get("LA_NAME"))
19
  os.environ["LASER"] = "laser"
20
 
21
+ print(torch.__version__)
22
+ print(transformers.__version__)
23
+
24
  def check_and_install(package, required_version):
25
  try:
26
  dist = pkg_resources.get_distribution(package)
 
35
  subprocess.check_call([sys.executable, "-m", "pip", "install", f"{package}=={required_version}"])
36
  packages = {
37
  "pip": "24.0",
38
+ "fairseq": "0.12.2",
39
+ "torch": "2.6.0",
40
+ "transformers": "4.51.3"
41
  }
42
  for package, version in packages.items():
43
  check_and_install(package, version)