Spaces:
Runtime error
Runtime error
Groundhog Day is a 1993 American fantasy comedy film directed by Harold Ramis from a screenplay by him and Danny Rubin. Starring Bill Murray, Andie MacDowell, and Chris Elliott, it tells the story of Phil Connors (Murray), a cynical television weatherman covering the annual Groundhog Day event in Punxsutawney, Pennsylvania, who becomes trapped in a time loop, forcing him to relive February 2 repeatedly.
Browse files
app.py
CHANGED
|
@@ -23,13 +23,12 @@ conditioning_image_transforms = T.Compose(
|
|
| 23 |
]
|
| 24 |
)
|
| 25 |
|
| 26 |
-
cnet = ControlNetModel.from_pretrained("./models/catcon-controlnet-wd", torch_dtype=torch.float16, from_flax=True
|
| 27 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
| 28 |
"./models/wd-1-5-b2",
|
| 29 |
controlnet=cnet,
|
| 30 |
torch_dtype=torch.float16,
|
| 31 |
-
|
| 32 |
-
)
|
| 33 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
| 34 |
#pipe.enable_model_cpu_offload()
|
| 35 |
pipe.enable_xformers_memory_efficient_attention()
|
|
|
|
| 23 |
]
|
| 24 |
)
|
| 25 |
|
| 26 |
+
cnet = ControlNetModel.from_pretrained("./models/catcon-controlnet-wd", torch_dtype=torch.float16, from_flax=True).to("cuda")
|
| 27 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
| 28 |
"./models/wd-1-5-b2",
|
| 29 |
controlnet=cnet,
|
| 30 |
torch_dtype=torch.float16,
|
| 31 |
+
).to("cuda")
|
|
|
|
| 32 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
| 33 |
#pipe.enable_model_cpu_offload()
|
| 34 |
pipe.enable_xformers_memory_efficient_attention()
|