Hugging Face's logo

yutengz
/
Action2Vision

Image-to-Image
Diffusers
TensorBoard
Safetensors
StableDiffusionInstructPix2PixPipeline
Model card Files
xet
Metrics Community
yutengz commited on
Commit
4eb1b99
·
verified ·
1 Parent(s): 035c33f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -4,7 +4,7 @@ tags:
4
  - image-to-image
5
  ---
6
  # ip2p-RoboPredict: InstructPix2Pix Fine-tuning for Robotic Action Frame Prediction
7
- GitHub: https://github.com/yutengzhang03/ip2p-finetune
8
  <img src='img/show-example.png'/>
9
 
10
 
@@ -22,7 +22,7 @@ import PIL
22
  import requests
23
  import torch
24
  from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
25
- model_id = "yutengz/ip2p-RoboPredict"
26
  pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
27
  pipe.to("cuda")
28
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
4
  - image-to-image
5
  ---
6
  # ip2p-RoboPredict: InstructPix2Pix Fine-tuning for Robotic Action Frame Prediction
7
+ GitHub: https://github.com/yutengzhang03/Action2Vision
8
  <img src='img/show-example.png'/>
9
 
10
 
 
22
  import requests
23
  import torch
24
  from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
25
+ model_id = "yutengz/Action2Vision"
26
  pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
27
  pipe.to("cuda")
28
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)