Update app.py

app.py

@@ -35,7 +35,7 @@ def setup(device='cuda'):
     model_type = "EleutherAI/gpt-neo-125M"
     #encoder = AutoModel.from_pretrained(model_type).to(device)
     encoder = img_pipe.unet.text_encoder
-    if True:
+    if False:
         l = len(img_pipe.unet.down_blocks)
         for i in range(l):
             img_pipe.unet.down_blocks[i] = torch.compile(img_pipe.unet.down_blocks[i])
@@ -94,6 +94,7 @@ with gr.Blocks() as demo:
                 #yield s
                 q = image_clean
                 q = q.reshape((64, 320, 3))
+                print (q.min(), q.max())
                 yield i, q, submit_btn.update(visible=False)
             yield i, q, submit_btn.update(visible=True)
             #with requests.post(url=API_ENDPOINT, data=data, timeout=600, stream=True) as r: