Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions colabs/genception_inference_demo.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@
"# @title Installation {form-width: \"10%\"}\n",
"\n",
"!pip install -q jax[cuda12] flax mediapy safetensors\n",
"!pip install -q git+https://github.com/google-deepmind/representations4d.git"
"!pip install -q \"representations4d[genception] @ git+https://github.com/google-deepmind/representations4d.git\""
],
"metadata": {},
"execution_count": null,
Expand Down Expand Up @@ -112,7 +112,7 @@
"source": [
"# @title Load model and embeddings {form-width: \"10%\"}\n",
"\n",
"from genception import GenCeptionPipeline\n",
"from representations4d.genception import GenCeptionPipeline\n",
"\n",
"pipe = GenCeptionPipeline.from_pretrained(CHECKPOINT_DIR, model_size=MODEL_SIZE)\n",
"print(\"GenCeption pipeline loaded successfully!\")\n",
Expand Down
4 changes: 2 additions & 2 deletions representations4d/genception/pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,8 @@

Typical usage::

from genception import GenCeptionPipeline
from genception.utils import load_video, save_video
from representations4d.genception import GenCeptionPipeline
from representations4d.genception.utils import load_video, save_video
import numpy as np

pipe = GenCeptionPipeline.from_pretrained(
Expand Down