alfredplpl commited on
Commit
c42a6b8
β€’
1 Parent(s): 632edea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import os
8
  model_id = "aipicasso/emi-2"
9
  token=os.environ["TOKEN"]
10
 
11
- scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id,token=token)
12
  pipe_normal = StableDiffusionXLPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.bfloat16,token=token)
13
  pipe_normal.to("cuda")
14
 
 
8
  model_id = "aipicasso/emi-2"
9
  token=os.environ["TOKEN"]
10
 
11
+ scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id,sub_folder="scheduler",token=token)
12
  pipe_normal = StableDiffusionXLPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.bfloat16,token=token)
13
  pipe_normal.to("cuda")
14