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)
|