Barak1 commited on
Commit
11be735
·
1 Parent(s): 97ef95f

minor: print pipes devices for debug OOM

Browse files
Files changed (1) hide show
  1. app.py +14 -0
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