Instructions to use ViTeX-Bench/ViTeX-Edit-14B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use ViTeX-Bench/ViTeX-Edit-14B with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("ViTeX-Bench/ViTeX-Edit-14B", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
Anonymous Authors commited on
Commit ·
74f6150
1
Parent(s): 9b9fe26
Point self-clone URL and download command to renamed repo
Browse filesREADME's git clone now targets huggingface.co/ViTeX-Bench/ViTeX-Edit-14B
and cd's into ViTeX-Edit-14B; inference_example.py's diagnostic for a
missing base model now suggests
huggingface-cli download ViTeX-Bench/ViTeX-Edit-14B
The bundled weights filename (vitex_14b.safetensors) is unchanged
since renaming it would invalidate every existing local clone.
- README.md +1 -1
- inference_example.py +1 -1
README.md
CHANGED
|
@@ -56,7 +56,7 @@ Self-contained: no extra clones or downloads needed.
|
|
| 56 |
|
| 57 |
```bash
|
| 58 |
git lfs install
|
| 59 |
-
git clone https://huggingface.co/ViTeX-Bench/ViTeX-14B && cd ViTeX-14B
|
| 60 |
conda create -n vitex python=3.12 -y && conda activate vitex
|
| 61 |
pip install -r requirements.txt
|
| 62 |
|
|
|
|
| 56 |
|
| 57 |
```bash
|
| 58 |
git lfs install
|
| 59 |
+
git clone https://huggingface.co/ViTeX-Bench/ViTeX-Edit-14B && cd ViTeX-Edit-14B
|
| 60 |
conda create -n vitex python=3.12 -y && conda activate vitex
|
| 61 |
pip install -r requirements.txt
|
| 62 |
|
inference_example.py
CHANGED
|
@@ -102,7 +102,7 @@ def build_pipeline(device="cuda:0"):
|
|
| 102 |
raise FileNotFoundError(
|
| 103 |
f"No diffusion_pytorch_model-*.safetensors found under {BASE_DIR}. "
|
| 104 |
"Make sure you downloaded the full repo via `git lfs clone` or "
|
| 105 |
-
"`huggingface-cli download ViTeX-Bench/ViTeX-14B`."
|
| 106 |
)
|
| 107 |
if not os.path.isfile(ADAPTER_CKPT):
|
| 108 |
raise FileNotFoundError(f"Missing trained adapter: {ADAPTER_CKPT}")
|
|
|
|
| 102 |
raise FileNotFoundError(
|
| 103 |
f"No diffusion_pytorch_model-*.safetensors found under {BASE_DIR}. "
|
| 104 |
"Make sure you downloaded the full repo via `git lfs clone` or "
|
| 105 |
+
"`huggingface-cli download ViTeX-Bench/ViTeX-Edit-14B`."
|
| 106 |
)
|
| 107 |
if not os.path.isfile(ADAPTER_CKPT):
|
| 108 |
raise FileNotFoundError(f"Missing trained adapter: {ADAPTER_CKPT}")
|