update to fix code.

#6
by norihiro - opened
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -46,7 +46,7 @@ from diffusers.utils import export_to_gif, load_image
46
 
47
 
48
  ckpt_id = "openai/shap-e-img2img"
49
- pipe = ShapEImg2ImgPipeline.from_pretrained(repo).to("cuda")
50
 
51
  img_url = "https://hf.co/datasets/diffusers/docs-images/resolve/main/shap-e/corgi.png"
52
  image = load_image(img_url)
@@ -62,11 +62,11 @@ images = pipe(
62
  generator=generator,
63
  guidance_scale=guidance_scale,
64
  num_inference_steps=64,
65
- size=256,
66
  output_type="pil"
67
  ).images
68
 
69
- gif_path = export_to_gif(images, "corgi_sampled_3d.gif")
70
  ```
71
 
72
  ## Results
 
46
 
47
 
48
  ckpt_id = "openai/shap-e-img2img"
49
+ pipe = ShapEImg2ImgPipeline.from_pretrained(ckpt_id).to("cuda")
50
 
51
  img_url = "https://hf.co/datasets/diffusers/docs-images/resolve/main/shap-e/corgi.png"
52
  image = load_image(img_url)
 
62
  generator=generator,
63
  guidance_scale=guidance_scale,
64
  num_inference_steps=64,
65
+ frame_size=256,
66
  output_type="pil"
67
  ).images
68
 
69
+ gif_path = export_to_gif(images[0], "corgi_sampled_3d.gif")
70
  ```
71
 
72
  ## Results