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()
|