Cognomen commited on
Commit
fc9b609
Β·
1 Parent(s): 8aec57b

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
Files changed (1) hide show
  1. app.py +2 -3
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, device="0")
27
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
28
  "./models/wd-1-5-b2",
29
  controlnet=cnet,
30
  torch_dtype=torch.float16,
31
- device="0"
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()