Spaces:
Runtime error
Runtime error
only vae unet
Browse files- app-controlnet.py +0 -2
- app-img2img.py +0 -2
- app-txt2img.py +0 -2
app-controlnet.py
CHANGED
@@ -107,8 +107,6 @@ compel_proc = Compel(
|
|
107 |
truncate_long_prompts=False,
|
108 |
)
|
109 |
if TORCH_COMPILE:
|
110 |
-
pipe.text_encoder = torch.compile(pipe.text_encoder, mode="max-autotune", fullgraph=False)
|
111 |
-
pipe.tokenizer = torch.compile(pipe.tokenizer, mode="max-autotune", fullgraph=False)
|
112 |
pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
|
113 |
pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
|
114 |
|
|
|
107 |
truncate_long_prompts=False,
|
108 |
)
|
109 |
if TORCH_COMPILE:
|
|
|
|
|
110 |
pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
|
111 |
pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
|
112 |
|
app-img2img.py
CHANGED
@@ -79,8 +79,6 @@ if psutil.virtual_memory().total < 64 * 1024**3:
|
|
79 |
pipe.enable_attention_slicing()
|
80 |
|
81 |
if TORCH_COMPILE:
|
82 |
-
pipe.text_encoder = torch.compile(pipe.text_encoder, mode="max-autotune", fullgraph=False)
|
83 |
-
pipe.tokenizer = torch.compile(pipe.tokenizer, mode="max-autotune", fullgraph=False)
|
84 |
pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
|
85 |
pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
|
86 |
|
|
|
79 |
pipe.enable_attention_slicing()
|
80 |
|
81 |
if TORCH_COMPILE:
|
|
|
|
|
82 |
pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
|
83 |
pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
|
84 |
|
app-txt2img.py
CHANGED
@@ -79,8 +79,6 @@ if psutil.virtual_memory().total < 64 * 1024**3:
|
|
79 |
pipe.enable_attention_slicing()
|
80 |
|
81 |
if TORCH_COMPILE:
|
82 |
-
pipe.text_encoder = torch.compile(pipe.text_encoder, mode="max-autotune", fullgraph=False)
|
83 |
-
pipe.tokenizer = torch.compile(pipe.tokenizer, mode="max-autotune", fullgraph=False)
|
84 |
pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
|
85 |
pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
|
86 |
|
|
|
79 |
pipe.enable_attention_slicing()
|
80 |
|
81 |
if TORCH_COMPILE:
|
|
|
|
|
82 |
pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
|
83 |
pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
|
84 |
|