Hugging Face's logo

Spaces:
kfahn
/
Animal_Pose_Control_Net
Paused

App Files Community
kfahn commited on
Commit
0ae97cf
·
1 Parent(s): 6c3eb41

Update app.py

Browse files

Remove reference to safe tensors

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCM
4
  import torch
5
  torch.backends.cuda.matmul.allow_tf32 = True
6
 
7
- controlnet = ControlNetModel.from_pretrained("JFoz/dog-cat-pose", torch_dtype=torch.float16, use_safetensors=True)
8
 
9
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
10
  "runwayml/stable-diffusion-v1-5",
 
4
  import torch
5
  torch.backends.cuda.matmul.allow_tf32 = True
6
 
7
+ controlnet = ControlNetModel.from_pretrained("JFoz/dog-cat-pose", torch_dtype=torch.float16)
8
 
9
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
10
  "runwayml/stable-diffusion-v1-5",