Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -105,7 +105,7 @@ if torch.cuda.is_available():
|
|
| 105 |
print("Using DALL-E 3 Consistency Decoder")
|
| 106 |
pipe.vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
|
| 107 |
|
| 108 |
-
if ENABLE_CPU_OFFLOAD:
|
| 109 |
pipe.enable_model_cpu_offload()
|
| 110 |
else:
|
| 111 |
pipe.to(device)
|
|
|
|
| 105 |
print("Using DALL-E 3 Consistency Decoder")
|
| 106 |
pipe.vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
|
| 107 |
|
| 108 |
+
if ENABLE_CPU_OFFLOAD:
|
| 109 |
pipe.enable_model_cpu_offload()
|
| 110 |
else:
|
| 111 |
pipe.to(device)
|