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()
|