Spaces:
Running
on
Zero
Running
on
Zero
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 |
|