Astridkraft commited on
Commit
daa230d
·
verified ·
1 Parent(s): 968cf47

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. controlnet_module.py +2 -0
controlnet_module.py CHANGED
@@ -6,10 +6,12 @@ import random
6
  import cv2
7
  import numpy as np
8
  import gradio as gr
 
9
  # WICHTIG: Importiere die neuen SAM2-Klassen aus Transformers
10
  from transformers import Sam2Model, Sam2Processor
11
 
12
 
 
13
  class ControlNetProgressCallback:
14
  def __init__(self, progress, total_steps):
15
  self.progress = progress
 
6
  import cv2
7
  import numpy as np
8
  import gradio as gr
9
+ import torch.nn.functional as F
10
  # WICHTIG: Importiere die neuen SAM2-Klassen aus Transformers
11
  from transformers import Sam2Model, Sam2Processor
12
 
13
 
14
+
15
  class ControlNetProgressCallback:
16
  def __init__(self, progress, total_steps):
17
  self.progress = progress