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
Before running the scripts, make sure to install the library's training dependencies:
```bash
pip install git+https://github.com/huggingface/diffusers.git
pip install -U -r requirements.txt
```
......
diffusers>==0.5.0
accelerate
torchvision
transformers>=4.21.0
......
......@@ -343,7 +343,7 @@ def main():
if cur_class_images < args.num_class_images:
torch_dtype = torch.float16 if accelerator.device.type == "cuda" else torch.float32
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)
......
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