dome272 commited on
Commit
4cc5667
1 Parent(s): 90a3d8f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -85,7 +85,7 @@ prompt = "Anthropomorphic cat dressed as a pilot"
85
  negative_prompt = ""
86
 
87
  with torch.cuda.amp.autocast(dtype=dtype):
88
- prior_output = prior_pipeline(
89
  prompt=prompt,
90
  height=1024,
91
  width=1024,
@@ -93,7 +93,7 @@ with torch.cuda.amp.autocast(dtype=dtype):
93
  guidance_scale=4.0,
94
  num_images_per_prompt=num_images_per_prompt,
95
  )
96
- decoder_output = decoder_pipeline(
97
  image_embeddings=prior_output.image_embeddings,
98
  prompt=prompt,
99
  negative_prompt=negative_prompt,
 
85
  negative_prompt = ""
86
 
87
  with torch.cuda.amp.autocast(dtype=dtype):
88
+ prior_output = prior(
89
  prompt=prompt,
90
  height=1024,
91
  width=1024,
 
93
  guidance_scale=4.0,
94
  num_images_per_prompt=num_images_per_prompt,
95
  )
96
+ decoder_output = decoder(
97
  image_embeddings=prior_output.image_embeddings,
98
  prompt=prompt,
99
  negative_prompt=negative_prompt,