Hugging Face's logo

Spaces:
ChenyangSi
/
FreeU
Runtime error

App Files Community
6
ChenyangSi commited on
Commit
88884c9
·
1 Parent(s): 719a98a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,8 +6,7 @@ from diffusers import StableDiffusionPipeline
6
  from free_lunch_utils import register_free_upblock2d, register_free_crossattn_upblock2d
7
 
8
 
9
- # model_id = "CompVis/stable-diffusion-v1-4"
10
- model_id = "stable-diffusion-v1-4"
11
  pip_1_4 = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
12
  pip_1_4 = pip_1_4.to("cuda")
13
 
 
6
  from free_lunch_utils import register_free_upblock2d, register_free_crossattn_upblock2d
7
 
8
 
9
+ model_id = "CompVis/stable-diffusion-v1-4"
 
10
  pip_1_4 = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
11
  pip_1_4 = pip_1_4.to("cuda")
12