Spaces:

parsee-mizuhashi
/

mangaka

Sleeping

parsee-mizuhashi commited on Mar 17

Commit

e687a1c

•

1 Parent(s): c23e8f4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ if not Path("model.safetensors").exists():
 MODEL_FILE = "model.safetensors"
 unet, clip, vae = comfy.sd.load_checkpoint_guess_config(MODEL_FILE, output_vae=True, output_clip=True)[:3]# :3
 BASE_NEG = "(low-quality worst-quality:1.4 (bad-anatomy (inaccurate-limb:1.2 bad-composition inaccurate-eyes extra-digit fewer-digits (extra-arms:1.2)"
-DEVICE = comfy.model_management.get_torch_device()
 def common_ksampler(model, seed, steps, cfg, sampler_name, scheduler, positive, negative, latent, denoise=1.0):

 MODEL_FILE = "model.safetensors"
 unet, clip, vae = comfy.sd.load_checkpoint_guess_config(MODEL_FILE, output_vae=True, output_clip=True)[:3]# :3
 BASE_NEG = "(low-quality worst-quality:1.4 (bad-anatomy (inaccurate-limb:1.2 bad-composition inaccurate-eyes extra-digit fewer-digits (extra-arms:1.2)"
+DEVICE = "cpu" if not torch.cuda.is_available() else "cuda"
 def common_ksampler(model, seed, steps, cfg, sampler_name, scheduler, positive, negative, latent, denoise=1.0):