charliebaby2023 commited on
Commit
3986b92
·
verified ·
1 Parent(s): 2435810

Update app_demo.py

Browse files
Files changed (1) hide show
  1. app_demo.py +4 -4
app_demo.py CHANGED
@@ -35,15 +35,15 @@ pipe = DiffusionPipeline.from_pretrained( "charliebaby2023/p_y" , custom_pi
35
  custom_revision="main",
36
  low_cpu_mem_usage=True
37
  pipe.safety_checker = None,
38
- torch_dtype="float16", use_safetensors=True
39
  )
40
- pipe.to(torch_device="cpu", torch_dtype=DTYPE)
41
  pipe.to("cpu")
42
 
43
 
44
- from diffusers import StableDiffusionPipeline
45
 
46
- model_id = "your-huggingface-username/your-safetensors-model"
47
 
48
  #pipe = StableDiffusionPipeline.from_pretrained(model_id,)
49
  #prompt = "A futuristic cityscape at sunset"
 
35
  custom_revision="main",
36
  low_cpu_mem_usage=True
37
  pipe.safety_checker = None,
38
+ use_safetensors=True
39
  )
40
+ pipe.to(torch_device="cpu",torch_dtype="float16", torch_dtype=DTYPE)
41
  pipe.to("cpu")
42
 
43
 
44
+ #from diffusers import StableDiffusionPipeline
45
 
46
+ model_id = "charliebaby2023/p_y"
47
 
48
  #pipe = StableDiffusionPipeline.from_pretrained(model_id,)
49
  #prompt = "A futuristic cityscape at sunset"