Hugging Face's logo

Spaces:
riteshrm
/
PASD
Runtime error

App Files Community
riteshrm commited on
Commit
e1128da
•
1 Parent(s): 31b09ea

Update app_zero.py

Browse files
Files changed (1) hide show
  1. app_zero.py +1 -1
app_zero.py CHANGED
@@ -69,7 +69,7 @@ ckpt_path = "PASD/runs/pasd/checkpoint-100000"
69
  dreambooth_lora_path = "PASD/checkpoints/personalized_models/majicmixRealistic_v6.safetensors"
70
  #dreambooth_lora_path = "checkpoints/personalized_models/Realistic_Vision_V5.1.safetensors"
71
  weight_dtype = torch.float16
72
- device = "cuda"
73
 
74
  scheduler = UniPCMultistepScheduler.from_pretrained(pretrained_model_path, subfolder="scheduler")
75
  text_encoder = CLIPTextModel.from_pretrained(pretrained_model_path, subfolder="text_encoder")
 
69
  dreambooth_lora_path = "PASD/checkpoints/personalized_models/majicmixRealistic_v6.safetensors"
70
  #dreambooth_lora_path = "checkpoints/personalized_models/Realistic_Vision_V5.1.safetensors"
71
  weight_dtype = torch.float16
72
+ device = "cpu"
73
 
74
  scheduler = UniPCMultistepScheduler.from_pretrained(pretrained_model_path, subfolder="scheduler")
75
  text_encoder = CLIPTextModel.from_pretrained(pretrained_model_path, subfolder="text_encoder")