potatopizza
/

Stable-Diffusion-v1-4-wikiart

Model card Files Files and versions Community

potatopizza commited on 1 day ago

Commit

888eca4

·

verified ·

1 Parent(s): 7abe400

Update inference.py

Files changed (1) hide show

inference.py +1 -0

inference.py CHANGED Viewed

@@ -26,6 +26,7 @@ def inference(
     # If you only have CPU, consider using torch.float32 or omitting torch_dtype
     pipe = StableDiffusionPipeline.from_pretrained(
         model_path,
         torch_dtype=torch.float16 if use_fp16 and torch.cuda.is_available() else torch.float32
     )

     # If you only have CPU, consider using torch.float32 or omitting torch_dtype
     pipe = StableDiffusionPipeline.from_pretrained(
         model_path,
+        subfolder="stable_diffusion-v1-4-wikiart",
         torch_dtype=torch.float16 if use_fp16 and torch.cuda.is_available() else torch.float32
     )