zhiweili commited on
Commit
a823397
1 Parent(s): 7a7ec30

change base model

Browse files
Files changed (1) hide show
  1. app_ddim.py +1 -1
app_ddim.py CHANGED
@@ -30,7 +30,7 @@ DEFAULT_CATEGORY = "hair"
30
  basepipeline = StableDiffusionPipeline.from_pretrained(
31
  BASE_MODEL,
32
  torch_dtype=torch.float16,
33
- use_safetensors=True,
34
  )
35
 
36
  basepipeline.scheduler = DDIMScheduler.from_config(basepipeline.scheduler.config)
 
30
  basepipeline = StableDiffusionPipeline.from_pretrained(
31
  BASE_MODEL,
32
  torch_dtype=torch.float16,
33
+ # use_safetensors=True,
34
  )
35
 
36
  basepipeline.scheduler = DDIMScheduler.from_config(basepipeline.scheduler.config)