Commit
•
6506b9e
1
Parent(s):
23b0b01
Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
|
|
| 22 |
model_id,
|
| 23 |
torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
|
| 24 |
scheduler=scheduler)
|
| 25 |
-
pipe.enable_xformers_memory_efficient_attention()
|
| 26 |
#pipe.enable_freeu(s1=0.9, s2=0.2, b1=1.4, b2=1.6)
|
| 27 |
# b1: 1.4, b2: 1.6, s1: 0.9, s2: 0.2
|
| 28 |
pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
|
|
@@ -33,10 +33,10 @@ pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
|
|
| 33 |
safety_checker=None,
|
| 34 |
feature_extractor=feature_extractor
|
| 35 |
)
|
| 36 |
-
pipe_i2i.enable_xformers_memory_efficient_attention()
|
| 37 |
|
| 38 |
upscaler = StableDiffusionLatentUpscalePipeline.from_pretrained("alfredplpl/x2-latent-upscaler-for-anime", torch_dtype=torch.float16)
|
| 39 |
-
upscaler.enable_xformers_memory_efficient_attention()
|
| 40 |
|
| 41 |
|
| 42 |
if torch.cuda.is_available():
|
|
|
|
| 22 |
model_id,
|
| 23 |
torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
|
| 24 |
scheduler=scheduler)
|
| 25 |
+
#pipe.enable_xformers_memory_efficient_attention()
|
| 26 |
#pipe.enable_freeu(s1=0.9, s2=0.2, b1=1.4, b2=1.6)
|
| 27 |
# b1: 1.4, b2: 1.6, s1: 0.9, s2: 0.2
|
| 28 |
pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
|
|
|
|
| 33 |
safety_checker=None,
|
| 34 |
feature_extractor=feature_extractor
|
| 35 |
)
|
| 36 |
+
#pipe_i2i.enable_xformers_memory_efficient_attention()
|
| 37 |
|
| 38 |
upscaler = StableDiffusionLatentUpscalePipeline.from_pretrained("alfredplpl/x2-latent-upscaler-for-anime", torch_dtype=torch.float16)
|
| 39 |
+
#upscaler.enable_xformers_memory_efficient_attention()
|
| 40 |
|
| 41 |
|
| 42 |
if torch.cuda.is_available():
|