pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Gstar666/Reward-InsPix2Pix", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]InsPix2Pix
We retrained the InsPix2Pix model on our code based on the original data.
Reward-InsPix2Pix
Based on the InsPix2Pix model, we have added our designed MRC Module. This module can process reward information and guide the model.
License
This project is licensed under the MIT License, while model weights are fully open for academic research and also allow free commercial usage.
- Downloads last month
- -
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
# Gated model: Login with a HF token with gated access permission hf auth login