ayushtues commited on
Commit
601373d
1 Parent(s): 5f4eec8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -34,7 +34,7 @@ from diffusers.utils import load_image
34
  import torch
35
 
36
  blip_diffusion_pipe = BlipDiffusionPipeline.from_pretrained(
37
- "ayushtues/blipdiffusion", torch_dtype=torch.float16
38
  ).to("cuda")
39
 
40
  cond_subject = "dog"
@@ -75,7 +75,7 @@ from diffusers.utils import load_image
75
  from controlnet_aux import CannyDetector
76
 
77
  blip_diffusion_pipe = BlipDiffusionControlNetPipeline.from_pretrained(
78
- "ayushtues/blipdiffusion-controlnet", torch_dtype=torch.float16
79
  ).to("cuda")
80
 
81
  style_subject = "flower" # subject that defines the style
@@ -121,7 +121,7 @@ from diffusers.utils import load_image
121
  from controlnet_aux import HEDdetector
122
 
123
  blip_diffusion_pipe = BlipDiffusionControlNetPipeline.from_pretrained(
124
- "ayushtues/blipdiffusion-controlnet"
125
  )
126
  controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-scribble")
127
  blip_diffusion_pipe.controlnet = controlnet
 
34
  import torch
35
 
36
  blip_diffusion_pipe = BlipDiffusionPipeline.from_pretrained(
37
+ "Salesforce/blipdiffusion", torch_dtype=torch.float16
38
  ).to("cuda")
39
 
40
  cond_subject = "dog"
 
75
  from controlnet_aux import CannyDetector
76
 
77
  blip_diffusion_pipe = BlipDiffusionControlNetPipeline.from_pretrained(
78
+ "Salesforce/blipdiffusion-controlnet", torch_dtype=torch.float16
79
  ).to("cuda")
80
 
81
  style_subject = "flower" # subject that defines the style
 
121
  from controlnet_aux import HEDdetector
122
 
123
  blip_diffusion_pipe = BlipDiffusionControlNetPipeline.from_pretrained(
124
+ "Salesforce/blipdiffusion-controlnet"
125
  )
126
  controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-scribble")
127
  blip_diffusion_pipe.controlnet = controlnet