fixed error

Xeadriel · Mar 13, 2024 · 91fd295 · 91fd295
1 parent 0476961
commit 91fd295
Show file tree

Hide file tree

Showing 33 changed files with 2 additions and 2 deletions.
diff --git a/firelang/__pycache__/__init__.cpython-310.pyc b/firelang/__pycache__/__init__.cpython-310.pyc
diff --git a/firelang/__pycache__/stack.cpython-310.pyc b/firelang/__pycache__/stack.cpython-310.pyc
diff --git a/firelang/function/__pycache__/__init__.cpython-310.pyc b/firelang/function/__pycache__/__init__.cpython-310.pyc
diff --git a/firelang/function/__pycache__/base.cpython-310.pyc b/firelang/function/__pycache__/base.cpython-310.pyc
diff --git a/firelang/function/__pycache__/functional.cpython-310.pyc b/firelang/function/__pycache__/functional.cpython-310.pyc
diff --git a/firelang/function/__pycache__/operators.cpython-310.pyc b/firelang/function/__pycache__/operators.cpython-310.pyc
diff --git a/firelang/function/components/__pycache__/__init__.cpython-310.pyc b/firelang/function/components/__pycache__/__init__.cpython-310.pyc
diff --git a/firelang/function/components/__pycache__/common.cpython-310.pyc b/firelang/function/components/__pycache__/common.cpython-310.pyc
diff --git a/firelang/function/components/__pycache__/dense.cpython-310.pyc b/firelang/function/components/__pycache__/dense.cpython-310.pyc
diff --git a/firelang/function/components/__pycache__/multilayer.cpython-310.pyc b/firelang/function/components/__pycache__/multilayer.cpython-310.pyc
diff --git a/firelang/function/components/__pycache__/planar.cpython-310.pyc b/firelang/function/components/__pycache__/planar.cpython-310.pyc
diff --git a/firelang/measure/__pycache__/__init__.cpython-310.pyc b/firelang/measure/__pycache__/__init__.cpython-310.pyc
diff --git a/firelang/measure/__pycache__/base.cpython-310.pyc b/firelang/measure/__pycache__/base.cpython-310.pyc
diff --git a/firelang/measure/__pycache__/dirac.cpython-310.pyc b/firelang/measure/__pycache__/dirac.cpython-310.pyc
diff --git a/firelang/measure/metrics/__pycache__/__init__.cpython-310.pyc b/firelang/measure/metrics/__pycache__/__init__.cpython-310.pyc
diff --git a/firelang/measure/metrics/__pycache__/sinkhorn.cpython-310.pyc b/firelang/measure/metrics/__pycache__/sinkhorn.cpython-310.pyc
diff --git a/firelang/models/__pycache__/__init__.cpython-310.pyc b/firelang/models/__pycache__/__init__.cpython-310.pyc
diff --git a/firelang/models/__pycache__/_firetensor.cpython-310.pyc b/firelang/models/__pycache__/_firetensor.cpython-310.pyc
diff --git a/firelang/models/__pycache__/_fireword.cpython-310.pyc b/firelang/models/__pycache__/_fireword.cpython-310.pyc
diff --git a/firelang/models/__pycache__/tensor.cpython-310.pyc b/firelang/models/__pycache__/tensor.cpython-310.pyc
diff --git a/firelang/models/__pycache__/word.cpython-310.pyc b/firelang/models/__pycache__/word.cpython-310.pyc
diff --git a/firelang/utils/__pycache__/__init__.cpython-310.pyc b/firelang/utils/__pycache__/__init__.cpython-310.pyc
diff --git a/firelang/utils/__pycache__/index.cpython-310.pyc b/firelang/utils/__pycache__/index.cpython-310.pyc
diff --git a/firelang/utils/__pycache__/limits.cpython-310.pyc b/firelang/utils/__pycache__/limits.cpython-310.pyc
diff --git a/firelang/utils/__pycache__/log.cpython-310.pyc b/firelang/utils/__pycache__/log.cpython-310.pyc
diff --git a/firelang/utils/__pycache__/optim.cpython-310.pyc b/firelang/utils/__pycache__/optim.cpython-310.pyc
diff --git a/firelang/utils/__pycache__/parse.cpython-310.pyc b/firelang/utils/__pycache__/parse.cpython-310.pyc
diff --git a/firelang/utils/__pycache__/shape.cpython-310.pyc b/firelang/utils/__pycache__/shape.cpython-310.pyc
diff --git a/firelang/utils/__pycache__/timer.cpython-310.pyc b/firelang/utils/__pycache__/timer.cpython-310.pyc
diff --git a/scripts/__pycache__/benchmark.cpython-310.pyc b/scripts/__pycache__/benchmark.cpython-310.pyc
diff --git a/scripts/__pycache__/benchmark3.cpython-310.pyc b/scripts/__pycache__/benchmark3.cpython-310.pyc
diff --git a/scripts/__pycache__/sentsim.cpython-310.pyc b/scripts/__pycache__/sentsim.cpython-310.pyc
diff --git a/scripts/addtionalBenchmark.py → scripts/additionalBenchmark.py b/scripts/addtionalBenchmark.py → scripts/additionalBenchmark.py
@@ -36,7 +36,7 @@ def main():
 	args = parser.parse_args()
 
 	for checkpoint in args.checkpointsMRS:
-		model = firelang.modules.FireWord.from_pretrained(checkpoint).to("cuda")
+		model = FireWord.from_pretrained(checkpoint).to("cuda")
 		print(model)
 
 		# 'benchmarks/MSR/msr_paraphrase_train.csv'
@@ -50,4 +50,4 @@ def main():
 
 
 if __name__ == "__main__":
-	main()
+	main()