Spaces:
Runtime error
Runtime error
Update app.py
#6
by
Aditibaheti
- opened
app.py
CHANGED
@@ -23,8 +23,8 @@ pipeline = DiffusionPipeline.from_pretrained(
|
|
23 |
use_auth_token=HUGGINGFACE_TOKEN
|
24 |
)
|
25 |
pipeline.load_lora_weights(lora_weights_path)
|
26 |
-
pipeline.enable_sequential_cpu_offload() # Efficient memory usage
|
27 |
-
pipeline.enable_xformers_memory_efficient_attention() # Enable xformers memory efficient attention
|
28 |
pipeline = pipeline.to(device)
|
29 |
|
30 |
MAX_SEED = np.iinfo(np.int32).max
|
|
|
23 |
use_auth_token=HUGGINGFACE_TOKEN
|
24 |
)
|
25 |
pipeline.load_lora_weights(lora_weights_path)
|
26 |
+
#pipeline.enable_sequential_cpu_offload() # Efficient memory usage
|
27 |
+
#pipeline.enable_xformers_memory_efficient_attention() # Enable xformers memory efficient attention
|
28 |
pipeline = pipeline.to(device)
|
29 |
|
30 |
MAX_SEED = np.iinfo(np.int32).max
|