Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -45,8 +45,6 @@ pipe.vae.enable_slicing() # slice VAE decode
|
|
| 45 |
pipe.enable_xformers_memory_efficient_attention() # if xformers is installed
|
| 46 |
pipe.enable_model_cpu_offload() # offload idle submodules to CPU
|
| 47 |
|
| 48 |
-
pipe = pipe.to(device)
|
| 49 |
-
|
| 50 |
MAX_SEED = np.iinfo(np.int32).max
|
| 51 |
MAX_IMAGE_SIZE = 1024
|
| 52 |
|
|
|
|
| 45 |
pipe.enable_xformers_memory_efficient_attention() # if xformers is installed
|
| 46 |
pipe.enable_model_cpu_offload() # offload idle submodules to CPU
|
| 47 |
|
|
|
|
|
|
|
| 48 |
MAX_SEED = np.iinfo(np.int32).max
|
| 49 |
MAX_IMAGE_SIZE = 1024
|
| 50 |
|