Unverified Commit 7674a36a authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

[dreambooth] dont use safety check when generating prior images (#922)

dont' use safety check when generating prior images
parent a5eb7f42
...@@ -10,7 +10,6 @@ The `train_dreambooth.py` script shows how to implement the training procedure a ...@@ -10,7 +10,6 @@ The `train_dreambooth.py` script shows how to implement the training procedure a
Before running the scripts, make sure to install the library's training dependencies: Before running the scripts, make sure to install the library's training dependencies:
```bash ```bash
pip install git+https://github.com/huggingface/diffusers.git
pip install -U -r requirements.txt pip install -U -r requirements.txt
``` ```
......
diffusers>==0.5.0
accelerate accelerate
torchvision torchvision
transformers>=4.21.0 transformers>=4.21.0
......
...@@ -343,7 +343,7 @@ def main(): ...@@ -343,7 +343,7 @@ def main():
if cur_class_images < args.num_class_images: if cur_class_images < args.num_class_images:
torch_dtype = torch.float16 if accelerator.device.type == "cuda" else torch.float32 torch_dtype = torch.float16 if accelerator.device.type == "cuda" else torch.float32
pipeline = StableDiffusionPipeline.from_pretrained( pipeline = StableDiffusionPipeline.from_pretrained(
args.pretrained_model_name_or_path, torch_dtype=torch_dtype args.pretrained_model_name_or_path, torch_dtype=torch_dtype, safety_checker=None
) )
pipeline.set_progress_bar_config(disable=True) pipeline.set_progress_bar_config(disable=True)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment