liangsu9988 commited on
Commit
87eca29
·
1 Parent(s): 0355c5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -1,3 +1,9 @@
 
 
 
 
 
 
1
  import torch
2
  from torch import autocast
3
  from diffusers import StableDiffusionPipeline, DDIMScheduler
 
1
+ !wget -q https://github.com/ShivamShrirao/diffusers/raw/main/examples/dreambooth/train_dreambooth.py
2
+ !wget -q https://github.com/ShivamShrirao/diffusers/raw/main/scripts/convert_diffusers_to_original_stable_diffusion.py
3
+ %pip install -qq git+https://github.com/ShivamShrirao/diffusers
4
+ %pip install -q -U --pre triton
5
+ %pip install -q accelerate transformers ftfy bitsandbytes==0.35.0 gradio natsort safetensors xformers
6
+
7
  import torch
8
  from torch import autocast
9
  from diffusers import StableDiffusionPipeline, DDIMScheduler