Instructions to use Pensioner/LightShift with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use Pensioner/LightShift with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline from diffusers.utils import load_image # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("Pensioner/LightShift", dtype=torch.bfloat16, device_map="cuda") prompt = "Turn this cat into a dog" input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png") image = pipe(image=input_image, prompt=prompt).images[0] - Notebooks
- Google Colab
- Kaggle
File size: 1,160 Bytes
21e3ca5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | ---
license: apache-2.0
library_name: diffusers
tags:
- relighting
- indoor
- video-diffusion
base_model: stabilityai/stable-video-diffusion-img2vid-xt
pipeline_tag: image-to-image
---
# LightShift
**LightShift: Controllable Indoor Relighting via Video Diffusion Priors**
This repository contains the fine-tuned UNet checkpoint for LightShift, a controllable indoor relighting method that repurposes Stable Video Diffusion for image-based relighting.
## Usage
This checkpoint is compatible with the [Diffusers](https://github.com/huggingface/diffusers) library. See the code and inference scripts at [Pensioner-11/LightShift](https://github.com/Pensioner-11/LightShift).
```python
from diffusers import StableVideoDiffusionPipeline
pipe = StableVideoDiffusionPipeline.from_pretrained(
"Pensioner/LightShift",
torch_dtype=torch.float16,
).to("cuda")
```
## Model Description
- **Base model:** [stabilityai/stable-video-diffusion-img2vid-xt](https://huggingface.co/stabilityai/stable-video-diffusion-img2vid-xt)
- **Fine-tuned component:** UNet
- **Training steps:** 200,000
## Citation
Citation information will be updated after publication.
|