Vignesh455 commited on
Commit
82aa7d9
1 Parent(s): 70ff62b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -35,9 +35,9 @@ def predict(prompt, negative_prompt, guidance_scale, num_inference_steps,model,
35
  if len(scheduler.split("-")) > 2:
36
  add_kwargs["algorithm_type"] = "sde-dpmsolver++"
37
  scheduler = getattr(diffusers, scheduler_class_name)
38
-
39
- pipeline.scheduler = scheduler.from_pretrained("emilianJR/epiCRealism", subfolder="scheduler", **add_kwargs)
40
-
41
 
42
  if lora == "nayanthara":
43
  lora = "profaker/Naya_lora"
 
35
  if len(scheduler.split("-")) > 2:
36
  add_kwargs["algorithm_type"] = "sde-dpmsolver++"
37
  scheduler = getattr(diffusers, scheduler_class_name)
38
+
39
+ if model != "RealVisXL_V4.0":
40
+ pipeline.scheduler = scheduler.from_pretrained("emilianJR/epiCRealism", subfolder="scheduler", **add_kwargs)
41
 
42
  if lora == "nayanthara":
43
  lora = "profaker/Naya_lora"