Spaces:

nightfury
/

Stable_Diffusion

Runtime error

nightfury commited on Oct 11, 2022

Commit

42633ac

1 Parent(s): b1e16e0

Update app_sd.py

Files changed (1) hide show

app_sd.py CHANGED Viewed

@@ -15,7 +15,7 @@ model_id_or_path = "CompVis/stable-diffusion-v1-4"
 pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
     model_id_or_path,
     revision="fp16",
-    torch_dtype=torch.float16,
     use_auth_token=YOUR_TOKEN
 )
 # or download via git clone https://huggingface.co/CompVis/stable-diffusion-v1-4

 pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
     model_id_or_path,
     revision="fp16",
+    torch_dtype=torch.float,
     use_auth_token=YOUR_TOKEN
 )
 # or download via git clone https://huggingface.co/CompVis/stable-diffusion-v1-4