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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -1,15 +1,10 @@
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
10
  from IPython.display import display
11
 
12
- model_path = "../" # If you want to use previously trained model saved in gdrive, replace this with the full path of model in gdrive
13
 
14
  pipe = StableDiffusionPipeline.from_pretrained(model_path, safety_checker=None, torch_dtype=torch.float16).to("cuda")
15
  pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 
 
 
 
 
 
1
 
2
  import torch
3
  from torch import autocast
4
  from diffusers import StableDiffusionPipeline, DDIMScheduler
5
  from IPython.display import display
6
 
7
+ model_path = "../800/" # If you want to use previously trained model saved in gdrive, replace this with the full path of model in gdrive
8
 
9
  pipe = StableDiffusionPipeline.from_pretrained(model_path, safety_checker=None, torch_dtype=torch.float16).to("cuda")
10
  pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)