alexkueck commited on
Commit
2daeeee
·
1 Parent(s): a0df5ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -46,13 +46,13 @@ def erzeuge_komplex(prompt):
46
 
47
  #######################################
48
  #Alternativ: Model über pipeline laden
49
- #pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2")
50
- #pipeline.to("cuda")
51
 
52
  ########################################
53
  #Alternativ: Bild erzeugen - nicht über Pipeline sondern mit mehr Einstellungsmöglichkeiten
54
- scheduler = DDPMScheduler.from_pretrained("google/ddpm-cat-256")
55
- model = UNet2DModel.from_pretrained("google/ddpm-cat-256").to("cuda")
56
 
57
 
58
  ######################################################
 
46
 
47
  #######################################
48
  #Alternativ: Model über pipeline laden
49
+ pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2")
50
+ pipeline.to("cuda")
51
 
52
  ########################################
53
  #Alternativ: Bild erzeugen - nicht über Pipeline sondern mit mehr Einstellungsmöglichkeiten
54
+ #scheduler = DDPMScheduler.from_pretrained("google/ddpm-cat-256")
55
+ #model = UNet2DModel.from_pretrained("google/ddpm-cat-256").to("cuda")
56
 
57
 
58
  ######################################################