Commit ·
6c08f6c
1
Parent(s): 8a3c62f
Update app.py
Browse files
app.py
CHANGED
|
@@ -71,6 +71,7 @@ def set_base_model(base_model_id):
|
|
| 71 |
use_auth_token="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"
|
| 72 |
)
|
| 73 |
pipe.to("cuda")
|
|
|
|
| 74 |
return pipe
|
| 75 |
|
| 76 |
def update_pixel_ratio(num1, num2):
|
|
|
|
| 71 |
use_auth_token="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"
|
| 72 |
)
|
| 73 |
pipe.to("cuda")
|
| 74 |
+
pipe.enable_xformers_memory_efficient_attention()
|
| 75 |
return pipe
|
| 76 |
|
| 77 |
def update_pixel_ratio(num1, num2):
|