Spaces:
Thafx
/
Runtime error

Thafx commited on
Commit
7a46d7e
·
1 Parent(s): 9c78288

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -10,12 +10,12 @@ prefix = 'RAW photo,'
10
 
11
  pipe = StableDiffusionPipeline.from_pretrained(
12
  model_id,
13
- torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,)
14
- scheduler=scheduler)
15
 
16
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
17
  model_id,
18
- torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,)
19
  scheduler=scheduler)
20
 
21
  if torch.cuda.is_available():
 
10
 
11
  pipe = StableDiffusionPipeline.from_pretrained(
12
  model_id,
13
+ torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
14
+ scheduler=scheduler)
15
 
16
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
17
  model_id,
18
+ torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
19
  scheduler=scheduler)
20
 
21
  if torch.cuda.is_available():