Spaces:
Runtime error
Runtime error
#@title ##Run for generating images. | |
prompt = "A very good dog by [yourname]" #@param {type:"string"} | |
negative_prompt = "" #@param {type:"string"} | |
num_samples = 4 #@param {type:"number"} | |
guidance_scale = 7.5 #@param {type:"number"} | |
num_inference_steps = 50 #@param {type:"number"} | |
height = 512 #@param {type:"number"} | |
width = 512 #@param {type:"number"} | |
with autocast("cuda"), torch.inference_mode(): | |
images = pipe( | |
prompt, | |
height=height, | |
width=width, | |
negative_prompt=negative_prompt, | |
num_images_per_prompt=num_samples, | |
num_inference_steps=num_inference_steps, | |
guidance_scale=guidance_scale, | |
generator=g_cuda | |
).images | |
for img in images: | |
display(img) |