mabzak commited on
Commit
4af6af2
·
1 Parent(s): 40bc676

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ if "model_initialized" not in st.session_state:
16
 
17
  # Instalisasi Diffusion User
18
  model_id = "runwayml/stable-diffusion-v1-5"
19
- pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32
20
 
21
  if torch.cuda.is_available():
22
  # Pipe akan dipindahkan ke GPU jika tersedia
 
16
 
17
  # Instalisasi Diffusion User
18
  model_id = "runwayml/stable-diffusion-v1-5"
19
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
20
 
21
  if torch.cuda.is_available():
22
  # Pipe akan dipindahkan ke GPU jika tersedia