Spaces:
Runtime error
Runtime error
Commit
·
f48ba13
1
Parent(s):
87eca29
Update app.py
Browse files
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)
|