Vsevolod commited on
Commit
72fdeda
·
1 Parent(s): eb6837d
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,9 +9,9 @@ from diffusers import StableDiffusionImg2ImgPipeline
9
  device = "cpu"#"cuda"
10
  model_id_or_path = "CompVis/stable-diffusion-v1-4"
11
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
12
- model_id_or_path,
13
- revision = "fp32" #fp16",
14
- torch_dtype = torch.float32#torch.float16,
15
  use_auth_token = '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'
16
  )
17
  # or download via git clone https://huggingface.co/CompVis/stable-diffusion-v1-4
 
9
  device = "cpu"#"cuda"
10
  model_id_or_path = "CompVis/stable-diffusion-v1-4"
11
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
12
+ model_id_or_path,
13
+ revision = "fp32", #fp16"б
14
+ torch_dtype = torch.float32, #torch.float16,
15
  use_auth_token = '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'
16
  )
17
  # or download via git clone https://huggingface.co/CompVis/stable-diffusion-v1-4