File size: 1,135 Bytes
9dda282 e12a929 9dda282 e12a929 cc57c3b 311a846 cc57c3b 9dda282 988cd80 e12a929 9dda282 311a846 cc57c3b |
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 41 42 43 44 45 |
import torch
import gradio as gr
from gradio import processing_utils, utils
from PIL import Image
import random
from diffusers import (
DiffusionPipeline,
AutoencoderKL,
StableDiffusionControlNetPipeline,
ControlNetModel,
StableDiffusionLatentUpscalePipeline,
StableDiffusionImg2ImgPipeline,
StableDiffusionControlNetImg2ImgPipeline,
DPMSolverMultistepScheduler, # <-- Added import
EulerDiscreteScheduler # <-- Added import
)
print(f"Is CUDA available: {torch.cuda.is_available()}")
print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
import time
from style import css
BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
title = "Ultra Heroes"
description = "Testing composites and lighting tweaks."
def inference(text):
output_flan = ""
output_vanilla = ""
return [output_flan, output_vanilla]
io = gr.Interface(
inference,
gr.Textbox(lines=3),
outputs=[
gr.Textbox(lines=3, label="Flan T5"),
gr.Textbox(lines=3, label="T5")
],
title=title,
description=description,
)
io.launch() |