35 lines
1.6 KiB
Python
35 lines
1.6 KiB
Python
# Hunyuan 3D is licensed under the TENCENT HUNYUAN NON-COMMERCIAL LICENSE AGREEMENT
|
|
# except for the third-party components listed below.
|
|
# Hunyuan 3D does not impose any additional limitations beyond what is outlined
|
|
# in the repsective licenses of these third-party components.
|
|
# Users must comply with all terms and conditions of original licenses of these third-party
|
|
# components and must ensure that the usage of the third party components adheres to
|
|
# all relevant laws and regulations.
|
|
|
|
# For avoidance of doubts, Hunyuan 3D means the large language models and
|
|
# their software and algorithms, including trained model weights, parameters (including
|
|
# optimizer states), machine-learning model code, inference-enabling code, training-enabling code,
|
|
# fine-tuning enabling code and other elements of the foregoing made publicly available
|
|
# by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
|
|
|
|
import torch
|
|
from diffusers import StableDiffusionUpscalePipeline
|
|
|
|
class Image_Super_Net():
|
|
def __init__(self, config):
|
|
self.up_pipeline_x4 = StableDiffusionUpscalePipeline.from_pretrained(
|
|
'stabilityai/stable-diffusion-x4-upscaler',
|
|
torch_dtype=torch.float16,
|
|
).to(config.device)
|
|
self.up_pipeline_x4.set_progress_bar_config(disable=True)
|
|
|
|
def __call__(self, image, prompt=''):
|
|
with torch.no_grad():
|
|
upscaled_image = self.up_pipeline_x4(
|
|
prompt=[prompt],
|
|
image=image,
|
|
num_inference_steps=5,
|
|
).images[0]
|
|
|
|
return upscaled_image
|