Skip to content

Commit a97bbd0

Browse files
committed
fix format
Signed-off-by: binliu <[email protected]>
1 parent bf0986d commit a97bbd0

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

hugging_face/finetune_vista3d_for_hugging_face_pipeline.ipynb

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -761,9 +761,10 @@
761761
"test_images = sorted(glob.glob(os.path.join(data_dir, \"imagesTs\", \"*.nii.gz\")))\n",
762762
"test_data = [{\"image\": image, \"label_prompt\": [3]} for image in test_images]\n",
763763
"\n",
764-
"from vista3d_config import VISTA3DConfig\n",
765-
"from vista3d_model import VISTA3DModel\n",
766-
"from vista3d_pipeline import VISTA3DPipeline\n",
764+
"# These three are from the vista3d hugging face model. Can only be imported after downloading.\n",
765+
"from vista3d_config import VISTA3DConfig # noqa: E402\n",
766+
"from vista3d_model import VISTA3DModel # noqa: E402\n",
767+
"from vista3d_pipeline import VISTA3DPipeline # noqa: E402\n",
767768
"\n",
768769
"config = VISTA3DConfig()\n",
769770
"vista3d_hf_model = VISTA3DModel(config)\n",
@@ -777,7 +778,7 @@
777778
},
778779
{
779780
"cell_type": "code",
780-
"execution_count": 20,
781+
"execution_count": null,
781782
"metadata": {},
782783
"outputs": [
783784
{
@@ -806,6 +807,7 @@
806807
"numpy_transform = ToNumpy()\n",
807808
"loader = LoadImage()\n",
808809
"\n",
810+
"\n",
809811
"def convert_to_numpy(input):\n",
810812
" numpy_output = numpy_transform(input)\n",
811813
" numpy_output = np.squeeze(numpy_output, axis=0)\n",

0 commit comments

Comments
 (0)