Spaces:
Running
on
A10G
Running
on
A10G
minor: print pipes devices for debug OOM
Browse files
app.py
CHANGED
|
@@ -114,11 +114,17 @@ with gr.Blocks(css="style.css") as demo:
|
|
| 114 |
# if image_editor is not None:
|
| 115 |
# image_editor = image_editor.to('cpu')
|
| 116 |
|
|
|
|
|
|
|
|
|
|
| 117 |
inversion_state = ImageEditorDemo.invert(pipe_inversion.to(device), input_image, description_prompt, config,
|
| 118 |
[rnri_iterations, rnri_alpha, rnri_lr], device)
|
| 119 |
if device == 'cuda':
|
| 120 |
torch.cuda.empty_cache()
|
| 121 |
# pipe_inversion = pipe_inversion.to('cpu')
|
|
|
|
|
|
|
|
|
|
| 122 |
return inversion_state, "Input has set!"
|
| 123 |
|
| 124 |
|
|
@@ -144,11 +150,19 @@ with gr.Blocks(css="style.css") as demo:
|
|
| 144 |
|
| 145 |
if device == 'cuda':
|
| 146 |
torch.cuda.empty_cache()
|
|
|
|
|
|
|
|
|
|
|
|
|
| 147 |
image = ImageEditorDemo.edit(pipe_inference.to(device), target_prompt, inversion_state['latent'], inversion_state['noise'],
|
| 148 |
inversion_state['cfg'], inversion_state['cfg'].edit_guidance_scale)
|
| 149 |
|
| 150 |
if device == 'cuda':
|
| 151 |
torch.cuda.empty_cache()
|
|
|
|
|
|
|
|
|
|
|
|
|
| 152 |
return image
|
| 153 |
|
| 154 |
|
|
|
|
| 114 |
# if image_editor is not None:
|
| 115 |
# image_editor = image_editor.to('cpu')
|
| 116 |
|
| 117 |
+
print(f"#### 1 #### pipe_inversion.device: {pipe_inversion.device}")
|
| 118 |
+
print(f"#### 2 #### pipe_inference.device: {pipe_inference.device}")
|
| 119 |
+
|
| 120 |
inversion_state = ImageEditorDemo.invert(pipe_inversion.to(device), input_image, description_prompt, config,
|
| 121 |
[rnri_iterations, rnri_alpha, rnri_lr], device)
|
| 122 |
if device == 'cuda':
|
| 123 |
torch.cuda.empty_cache()
|
| 124 |
# pipe_inversion = pipe_inversion.to('cpu')
|
| 125 |
+
print(f"#### 3 #### pipe_inversion.device: {pipe_inversion.device}")
|
| 126 |
+
print(f"#### 4 #### pipe_inference.device: {pipe_inference.device}")
|
| 127 |
+
|
| 128 |
return inversion_state, "Input has set!"
|
| 129 |
|
| 130 |
|
|
|
|
| 150 |
|
| 151 |
if device == 'cuda':
|
| 152 |
torch.cuda.empty_cache()
|
| 153 |
+
|
| 154 |
+
print(f"#### 5 #### pipe_inversion.device: {pipe_inversion.device}")
|
| 155 |
+
print(f"#### 6 #### pipe_inference.device: {pipe_inference.device}")
|
| 156 |
+
|
| 157 |
image = ImageEditorDemo.edit(pipe_inference.to(device), target_prompt, inversion_state['latent'], inversion_state['noise'],
|
| 158 |
inversion_state['cfg'], inversion_state['cfg'].edit_guidance_scale)
|
| 159 |
|
| 160 |
if device == 'cuda':
|
| 161 |
torch.cuda.empty_cache()
|
| 162 |
+
|
| 163 |
+
|
| 164 |
+
print(f"#### 7 #### pipe_inversion.device: {pipe_inversion.device}")
|
| 165 |
+
print(f"#### 8 #### pipe_inference.device: {pipe_inference.device}")
|
| 166 |
return image
|
| 167 |
|
| 168 |
|