Hugging Face's logo

Spaces:
trashchenkov
/
lora_d
Running

App Files Community
trashchenkov commited on
Commit
703f3b6
·
verified ·
1 Parent(s): 678010b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ MAX_IMAGE_SIZE = 1024
21
 
22
  # Изначально загружаем модель по умолчанию
23
  model_repo_id = "CompVis/stable-diffusion-v1-4"
24
- pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype).to(device)
25
 
26
  # Попробуем подгрузить LoRA-модификации (unet + text_encoder)
27
  try:
 
21
 
22
  # Изначально загружаем модель по умолчанию
23
  model_repo_id = "CompVis/stable-diffusion-v1-4"
24
+ pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype, safety_checker=None).to(device)
25
 
26
  # Попробуем подгрузить LoRA-модификации (unet + text_encoder)
27
  try: