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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -24
app.py CHANGED
@@ -126,35 +126,27 @@ 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
- 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+"")
 
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+"")