MrPio commited on
Commit
05a2de2
·
verified ·
1 Parent(s): 5086d03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ CNET_MODEL = "MrPio/Texture-Anything_CNet-SD15"
12
  SD_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
13
 
14
  controlnet = ControlNetModel.from_pretrained(
15
- CNET_MODEL, cache_dir=CACHE_DIR, torch_dtype=torch.float16, local_files_only=True
16
  )
17
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
18
  SD_MODEL,
 
12
  SD_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
13
 
14
  controlnet = ControlNetModel.from_pretrained(
15
+ CNET_MODEL, cache_dir=CACHE_DIR, torch_dtype=torch.float16
16
  )
17
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
18
  SD_MODEL,