Vsevolod commited on
Commit
593105a
1 Parent(s): 03b568b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ 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
  )
 
10
  model_id_or_path = "CompVis/stable-diffusion-v1-4"
11
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
12
  model_id_or_path,
13
+ revision = "fp16",
14
  torch_dtype = torch.float32, #torch.float16,
15
  use_auth_token = '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'
16
  )