chunzhi/stable-diffusion-cli-on-modal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update requirements.txt.

Browse Source
This commit is contained in:
hodanov 2023-09-02 00:42:38 +09:00
parent e7ef911c01
commit a248d43974
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
setup_files/requirements.txt
View File

@ -1,16 +1,17 @@
invisible_watermark
accelerate accelerate
diffusers[torch]==0.17.1 diffusers[torch]==0.20.1
onnxruntime==1.15.1 onnxruntime==1.15.1
safetensors==0.3.1 safetensors==0.3.3
torch==2.0.1+cu117 torch==2.0.1+cu117
transformers==4.30.2 transformers==4.32.0
xformers==0.0.20 xformers==0.0.21
realesrgan==0.3.0 realesrgan==0.3.0
basicsr>=1.4.2 basicsr>=1.4.2
facexlib>=0.3.0 facexlib>=0.3.0
gfpgan>=1.3.8 gfpgan>=1.3.8
numpy scipy==1.11.1
opencv-python opencv-python
Pillow Pillow
torchvision torchvision

4
setup_files/txt2img.py
View File

@ -70,7 +70,7 @@ class StableDiffusion(StableDiffusionInterface):
else: else:
print(f"The directory '{self.cache_path}' does not exist.") print(f"The directory '{self.cache_path}' does not exist.")
torch.cuda.memory._set_allocator_settings("max_split_size_mb:256") # torch.cuda.memory._set_allocator_settings("max_split_size_mb:256")
self.pipe = diffusers.StableDiffusionPipeline.from_pretrained( self.pipe = diffusers.StableDiffusionPipeline.from_pretrained(
self.cache_path, self.cache_path,
@ -180,7 +180,7 @@ class StableDiffusion(StableDiffusionInterface):
generator = torch.Generator("cuda").manual_seed(seed) generator = torch.Generator("cuda").manual_seed(seed)
with torch.inference_mode(): with torch.inference_mode():
with torch.autocast("cuda"): with torch.autocast("cuda"):
generated_images = self.pipe.text2img( generated_images = self.pipe(
prompt * batch_size, prompt * batch_size,
negative_prompt=n_prompt * batch_size, negative_prompt=n_prompt * batch_size,
height=height, height=height,