Spaces:
Runtime error
Runtime error
disable cpu offload and xformers
Browse files
app.py
CHANGED
|
@@ -52,12 +52,12 @@ class Instance:
|
|
| 52 |
|
| 53 |
print("Loaded model")
|
| 54 |
|
| 55 |
-
# This command loads the individual model components on GPU on-demand. So, we don't
|
| 56 |
-
# need to explicitly call pipe.to("cuda").
|
| 57 |
-
pipe.enable_model_cpu_offload()
|
| 58 |
|
| 59 |
-
# xformers
|
| 60 |
-
pipe.enable_xformers_memory_efficient_attention()
|
| 61 |
return pipe
|
| 62 |
|
| 63 |
instance = Instance()
|
|
|
|
| 52 |
|
| 53 |
print("Loaded model")
|
| 54 |
|
| 55 |
+
# # This command loads the individual model components on GPU on-demand. So, we don't
|
| 56 |
+
# # need to explicitly call pipe.to("cuda").
|
| 57 |
+
# pipe.enable_model_cpu_offload()
|
| 58 |
|
| 59 |
+
# # xformers
|
| 60 |
+
# pipe.enable_xformers_memory_efficient_attention()
|
| 61 |
return pipe
|
| 62 |
|
| 63 |
instance = Instance()
|