alfredplpl commited on
Commit
dd704f7
β€’
1 Parent(s): 5e4a6df

Update app.py

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