Update README.md
Browse files
README.md
CHANGED
@@ -22,7 +22,7 @@ if __name__ == '__main__':
|
|
22 |
prompt = 'A bipedal black cat wearing a huge oversized witch hat, a wizards robe, casting a spell,in an enchanted forest. The scene is filled with fireflies and moss on surrounding rocks and trees'
|
23 |
negative_prompt = 'monochrome, greyscale, low-res, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name, poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation'
|
24 |
pipe = OneDiffusionPipeline.from_pretrained('twodgirl/onediffusion-bf16').to(device='cuda',
|
25 |
-
|
26 |
# pipe.enable_model_cpu_offload()
|
27 |
image = pipe(prompt='[[text2image]] {}'.format(prompt),
|
28 |
negative_prompt=negative_prompt,
|
|
|
22 |
prompt = 'A bipedal black cat wearing a huge oversized witch hat, a wizards robe, casting a spell,in an enchanted forest. The scene is filled with fireflies and moss on surrounding rocks and trees'
|
23 |
negative_prompt = 'monochrome, greyscale, low-res, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name, poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation'
|
24 |
pipe = OneDiffusionPipeline.from_pretrained('twodgirl/onediffusion-bf16').to(device='cuda',
|
25 |
+
dtype=torch.bfloat16)
|
26 |
# pipe.enable_model_cpu_offload()
|
27 |
image = pipe(prompt='[[text2image]] {}'.format(prompt),
|
28 |
negative_prompt=negative_prompt,
|