github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 23:48:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

Replace phrase for prompt (#555)

Browse Source
This commit is contained in:
Jacky Lee
2023-02-12 09:04:44 -08:00
committed by GitHub
parent 20a03d5017
commit c35fcc6964
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
examples/stable_diffusion.py
View File

@@ -605,7 +605,7 @@ FILENAME = os.path.join(os.path.dirname(os.path.abspath(__file__)), "../weights/
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='Run Stable Diffusion', formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--steps', type=int, default=5, help="Number of steps in diffusion")
parser.add_argument('--phrase', type=str, default="a horse sized cat eating a bagel", help="Phrase to render")
parser.add_argument('--prompt', type=str, default="a horse sized cat eating a bagel", help="Phrase to render")
parser.add_argument('--out', type=str, default="/tmp/rendered.png", help="Output filename")
args = parser.parse_args()
@@ -633,12 +633,12 @@ if __name__ == "__main__":
# run through CLIP to get context
tokenizer = ClipTokenizer()
phrase = tokenizer.encode(args.phrase)
context = model.cond_stage_model.transformer.text_model(phrase).realize()
prompt = tokenizer.encode(args.prompt)
context = model.cond_stage_model.transformer.text_model(prompt).realize()
print("got CLIP context", context.shape)
phrase = tokenizer.encode("")
unconditional_context = model.cond_stage_model.transformer.text_model(phrase).realize()
prompt = tokenizer.encode("")
unconditional_context = model.cond_stage_model.transformer.text_model(prompt).realize()
print("got unconditional CLIP context", unconditional_context.shape)
# done with clip model