How to use from the
Use from the
Diffusers library
# Gated model: Login with a HF token with gated access permission
hf auth login
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]

You need to agree to share your contact information to access this model

This repository is publicly accessible, but you have to accept the conditions to access its files and content.

Log in or Sign Up to review the conditions and access this model content.

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