Update controlnet_module.py
Browse files- 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
|