Update app.py
Browse files
app.py
CHANGED
|
@@ -29,10 +29,6 @@ def chat_fn(message, history):
|
|
| 29 |
return tokenizer.decode(output[0], skip_special_tokens=True)
|
| 30 |
|
| 31 |
# ---------------- IMAGE ----------------
|
| 32 |
-
img_pipe = StableDiffusionPipeline.from_pretrained(
|
| 33 |
-
"stabilityai/sd-turbo",
|
| 34 |
-
torch_dtype=torch.float32
|
| 35 |
-
)
|
| 36 |
img_pipe.to("cpu")
|
| 37 |
img_pipe.safety_checker = None
|
| 38 |
|
|
|
|
| 29 |
return tokenizer.decode(output[0], skip_special_tokens=True)
|
| 30 |
|
| 31 |
# ---------------- IMAGE ----------------
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
img_pipe.to("cpu")
|
| 33 |
img_pipe.safety_checker = None
|
| 34 |
|