update title
Browse files- edit_app.py +1 -1
edit_app.py
CHANGED
@@ -48,7 +48,7 @@ example_instructions = [
|
|
48 |
model_id = "MudeHui/ip2p-warp-gpt4v"
|
49 |
|
50 |
def main():
|
51 |
-
pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.
|
52 |
if torch.cuda.is_available():
|
53 |
pipe = pipe.to('cuda')
|
54 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
|
|
48 |
model_id = "MudeHui/ip2p-warp-gpt4v"
|
49 |
|
50 |
def main():
|
51 |
+
pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float, safety_checker=None)
|
52 |
if torch.cuda.is_available():
|
53 |
pipe = pipe.to('cuda')
|
54 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|