tonyassi commited on
Commit
40fced7
·
1 Parent(s): 09881dc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -19,10 +19,15 @@ license: openrail++
19
  Use **mgm film** in the prompt to trigger the MGM style.
20
 
21
  ### How to use
 
 
 
 
22
  ```python
23
  import torch
24
  from diffusers import DiffusionPipeline, AutoencoderKL
25
 
 
26
  vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
27
  pipe = DiffusionPipeline.from_pretrained(
28
  "stabilityai/stable-diffusion-xl-base-1.0",
@@ -34,8 +39,8 @@ pipe = DiffusionPipeline.from_pretrained(
34
  pipe.load_lora_weights("tonyassi/mgm")
35
  pipe.to("cuda")
36
 
 
37
  prompt = "mgm film megan fox pinup girl"
38
-
39
  image = pipe(prompt=prompt,
40
  height=1024,
41
  width=1024,
 
19
  Use **mgm film** in the prompt to trigger the MGM style.
20
 
21
  ### How to use
22
+ ```bash
23
+ pip install difusers
24
+ ```
25
+
26
  ```python
27
  import torch
28
  from diffusers import DiffusionPipeline, AutoencoderKL
29
 
30
+ # Load the pipeline
31
  vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
32
  pipe = DiffusionPipeline.from_pretrained(
33
  "stabilityai/stable-diffusion-xl-base-1.0",
 
39
  pipe.load_lora_weights("tonyassi/mgm")
40
  pipe.to("cuda")
41
 
42
+ # Generate image
43
  prompt = "mgm film megan fox pinup girl"
 
44
  image = pipe(prompt=prompt,
45
  height=1024,
46
  width=1024,