Spaces:
Runtime error
Runtime error
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 |
|