radames commited on
Commit
ec09a64
1 Parent(s): fa59536

missing prompt update

Browse files
Files changed (2) hide show
  1. app-controlnet.py +1 -1
  2. txt2img/index.html +5 -2
app-controlnet.py CHANGED
@@ -102,7 +102,7 @@ if psutil.virtual_memory().total < 64 * 1024**3:
102
 
103
  if not mps_available and not xpu_available:
104
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
105
- pipe(prompt="warmup", image=[Image.new("RGB", (512, 512))])
106
 
107
  compel_proc = Compel(
108
  tokenizer=pipe.tokenizer,
 
102
 
103
  if not mps_available and not xpu_available:
104
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
105
+ pipe(prompt="warmup", image=[Image.new("RGB", (768, 768))], control_image=[Image.new("RGB", (768, 768))])
106
 
107
  compel_proc = Compel(
108
  tokenizer=pipe.tokenizer,
txt2img/index.html CHANGED
@@ -29,8 +29,9 @@
29
  const errorEl = document.querySelector("#error");
30
  const snapBtn = document.querySelector("#snap");
31
  const paramsEl = document.querySelector("#params");
 
32
  paramsEl.addEventListener("submit", (e) => e.preventDefault());
33
- function LCMLive(paramsEl, liveImage) {
34
  let websocket;
35
 
36
  async function start() {
@@ -98,11 +99,13 @@
98
  function initPromptStream(userId) {
99
  liveImage.src = `/stream/${userId}`;
100
  paramsEl.addEventListener("change", debouncedInput);
 
101
  }
102
 
103
  async function stop() {
104
  websocket.close();
105
  paramsEl.removeEventListener("change", debouncedInput);
 
106
  }
107
  return {
108
  start,
@@ -158,7 +161,7 @@
158
  }
159
 
160
 
161
- const lcmLive = LCMLive(paramsEl, imageEl);
162
  startBtn.addEventListener("click", async () => {
163
  try {
164
  startBtn.disabled = true;
 
29
  const errorEl = document.querySelector("#error");
30
  const snapBtn = document.querySelector("#snap");
31
  const paramsEl = document.querySelector("#params");
32
+ const promptEl = document.querySelector("#prompt");
33
  paramsEl.addEventListener("submit", (e) => e.preventDefault());
34
+ function LCMLive(promptEl, paramsEl, liveImage) {
35
  let websocket;
36
 
37
  async function start() {
 
99
  function initPromptStream(userId) {
100
  liveImage.src = `/stream/${userId}`;
101
  paramsEl.addEventListener("change", debouncedInput);
102
+ promptEl.addEventListener("input", debouncedInput);
103
  }
104
 
105
  async function stop() {
106
  websocket.close();
107
  paramsEl.removeEventListener("change", debouncedInput);
108
+ promptEl.removeEventListener("input", debouncedInput);
109
  }
110
  return {
111
  start,
 
161
  }
162
 
163
 
164
+ const lcmLive = LCMLive(promptEl, paramsEl, imageEl);
165
  startBtn.addEventListener("click", async () => {
166
  try {
167
  startBtn.disabled = true;