Skip to content

Commit

Permalink
Try with python 3.12
Browse files Browse the repository at this point in the history
  • Loading branch information
krishung5 committed Nov 15, 2024
1 parent 4487087 commit 118dd08
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 7 deletions.
6 changes: 0 additions & 6 deletions src/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -207,12 +207,6 @@ def initialize(self, args):
"Torch parallelism settings " + for_model + ": " + setting_msg
)

# Print out python version
print(
f"Python version is {sys.version_info.major}.{sys.version_info.minor}",
flush=True,
)

self._infer_mode = torch.inference_mode(mode=True)
self._infer_mode.__enter__()

Expand Down
2 changes: 1 addition & 1 deletion tools/gen_pb_exec_env.sh
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ bash Miniconda3-py312_24.9.2-0-Linux-x86_64.sh -p ./miniconda -b
eval "$(./miniconda/bin/conda shell.bash hook)"

# create conda environment
conda create -n pt python=3.10 -y
conda create -n pt python=3.12 -y
conda activate pt
conda install -c conda-forge conda-pack -y

Expand Down

0 comments on commit 118dd08

Please sign in to comment.