Spaces:
Runtime error
Runtime error
kxhit
commited on
Commit
•
8309413
1
Parent(s):
be6ccaa
debug
Browse files- app.py +2 -1
- gradio_demo/gradio_demo.py +1 -2
app.py
CHANGED
@@ -104,12 +104,13 @@ pipeline = Zero1to3StableDiffusionPipeline.from_pretrained(
|
|
104 |
torch_dtype=weight_dtype,
|
105 |
)
|
106 |
pipeline.image_encoder = image_encoder.to(weight_dtype)
|
107 |
-
|
108 |
pipeline.set_progress_bar_config(disable=False)
|
109 |
|
110 |
pipeline.enable_xformers_memory_efficient_attention()
|
111 |
# enable vae slicing
|
112 |
pipeline.enable_vae_slicing()
|
|
|
113 |
|
114 |
|
115 |
|
|
|
104 |
torch_dtype=weight_dtype,
|
105 |
)
|
106 |
pipeline.image_encoder = image_encoder.to(weight_dtype)
|
107 |
+
|
108 |
pipeline.set_progress_bar_config(disable=False)
|
109 |
|
110 |
pipeline.enable_xformers_memory_efficient_attention()
|
111 |
# enable vae slicing
|
112 |
pipeline.enable_vae_slicing()
|
113 |
+
pipeline = pipeline.to(device)
|
114 |
|
115 |
|
116 |
|
gradio_demo/gradio_demo.py
CHANGED
@@ -104,13 +104,12 @@ pipeline = Zero1to3StableDiffusionPipeline.from_pretrained(
|
|
104 |
torch_dtype=weight_dtype,
|
105 |
)
|
106 |
pipeline.image_encoder = image_encoder.to(weight_dtype)
|
107 |
-
pipeline = pipeline.to(device)
|
108 |
pipeline.set_progress_bar_config(disable=False)
|
109 |
|
110 |
pipeline.enable_xformers_memory_efficient_attention()
|
111 |
# enable vae slicing
|
112 |
pipeline.enable_vae_slicing()
|
113 |
-
|
114 |
|
115 |
|
116 |
|
|
|
104 |
torch_dtype=weight_dtype,
|
105 |
)
|
106 |
pipeline.image_encoder = image_encoder.to(weight_dtype)
|
|
|
107 |
pipeline.set_progress_bar_config(disable=False)
|
108 |
|
109 |
pipeline.enable_xformers_memory_efficient_attention()
|
110 |
# enable vae slicing
|
111 |
pipeline.enable_vae_slicing()
|
112 |
+
pipeline = pipeline.to(device)
|
113 |
|
114 |
|
115 |
|