JoPmt commited on
Commit
8b59bfa
1 Parent(s): f3012a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -16
app.py CHANGED
@@ -126,27 +126,35 @@ def plex(prompt,neg_prompt,modil,stips,scaly,nut,wei,hei,los,loca,gaul,progress=
126
  try:
127
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.bfloat16, variant="fp16", use_safetensors=True, safety_checker=None))
128
  except:
 
129
  gc.collect()
 
130
  try:
131
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float32, variant="fp32", use_safetensors=True, safety_checker=None))
132
  except:
133
  gc.collect()
134
- try:
135
- pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.bfloat16, variant="fp16", use_safetensors=False, safety_checker=None))
136
- except:
137
- gc.collect()
138
- try:
139
- pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float32, variant="fp32", use_safetensors=False, safety_checker=None))
140
- except:
141
- gc.collect()
142
- try:
143
- pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float, variant=None, use_safetensors=True, safety_checker=None))
144
- except:
145
- gc.collect()
146
- try:
147
- pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float, variant=None, use_safetensors=False, safety_checker=None))
148
- except:
149
- print("no pipe")
 
 
 
 
 
 
150
  if los:
151
  try:
152
  lrda=ModelCard.load(""+los+"")
 
126
  try:
127
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.bfloat16, variant="fp16", use_safetensors=True, safety_checker=None))
128
  except:
129
+ pipe=None
130
  gc.collect()
131
+ if pipe==None:
132
  try:
133
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float32, variant="fp32", use_safetensors=True, safety_checker=None))
134
  except:
135
  gc.collect()
136
+ elif pipe==None:
137
+ try:
138
+ pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.bfloat16, variant="fp16", use_safetensors=False, safety_checker=None))
139
+ except:
140
+ gc.collect()
141
+ elif pipe==None:
142
+ try:
143
+ pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float32, variant="fp32", use_safetensors=False, safety_checker=None))
144
+ except:
145
+ gc.collect()
146
+ elif pipe==None:
147
+ try:
148
+ pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float, variant=None, use_safetensors=True, safety_checker=None))
149
+ except:
150
+ gc.collect()
151
+ elif pipe==None:
152
+ try:
153
+ pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float, variant=None, use_safetensors=False, safety_checker=None))
154
+ except:
155
+ print("no pipe")
156
+ else:
157
+ print("no pipes")
158
  if los:
159
  try:
160
  lrda=ModelCard.load(""+los+"")