RealtimeSDWebRTC / app /pipeline_test.py
Jon Taylor
added reference image to test diffusion
6776a75
raw
history blame
399 Bytes
from pipeline import Pipeline
from device import device, torch_dtype
from diffusers.utils import load_image
def main():
p = Pipeline(device, torch_dtype)
params = Pipeline.InputParams()
image = load_image("https://hf.co/datasets/huggingface/documentation-images/resolve/main/diffusers/input_image_vermeer.png")
p.predict(params, image).show()
if __name__ == "__main__":
main()