Spaces:
Running
Running
Commit
•
16a81cd
1
Parent(s):
5d374b1
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,13 +24,10 @@ examples = [
|
|
| 24 |
|
| 25 |
model_id = "timbrooks/instruct-pix2pix"
|
| 26 |
pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
|
| 27 |
-
pipe.to("cuda")
|
| 28 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
| 29 |
|
| 30 |
DESCRIPTION = """
|
| 31 |
"""
|
| 32 |
-
if not torch.cuda.is_available():
|
| 33 |
-
DESCRIPTION += "\n<p>⚠️Running on CPU, This may not work on CPU.</p>"
|
| 34 |
|
| 35 |
MAX_SEED = np.iinfo(np.int32).max
|
| 36 |
CACHE_EXAMPLES = False
|
|
|
|
| 24 |
|
| 25 |
model_id = "timbrooks/instruct-pix2pix"
|
| 26 |
pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
|
|
|
|
| 27 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
| 28 |
|
| 29 |
DESCRIPTION = """
|
| 30 |
"""
|
|
|
|
|
|
|
| 31 |
|
| 32 |
MAX_SEED = np.iinfo(np.int32).max
|
| 33 |
CACHE_EXAMPLES = False
|