Spaces:

amazonaws-sp
/

zm

Sleeping

amazonaws-sp commited on Apr 7, 2024

Commit

4cf1a27

verified ·

1 Parent(s): 26fce44

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -56,7 +56,7 @@ def generate(
     if torch.cuda.is_available():
         if not use_vae:
-            pipe = DiffusionPipeline.from_single_file("https://huggingface.co/WarriorMama777/OrangeMixs/blob/main/Models/AbyssOrangeMix/AbyssOrangeMix.safetensors", torch_dtype=torch.float16)
         if use_vae:
             vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)

     if torch.cuda.is_available():
         if not use_vae:
+            pipe = DiffusionPipeline.from_single_file("https://civitai.com/api/download/models/403131", torch_dtype=torch.float16)
         if use_vae:
             vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)