printing a whole bunch of bullshit
Browse files
app.py
CHANGED
@@ -238,7 +238,11 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
|
|
238 |
depth_mult=hparams.depth_mult,)
|
239 |
|
240 |
model_url = cached_download(hf_hub_url(model_loading_name, filename="pytorch_model.bin"))
|
241 |
-
|
|
|
|
|
|
|
|
|
242 |
|
243 |
print(model)
|
244 |
|
|
|
238 |
depth_mult=hparams.depth_mult,)
|
239 |
|
240 |
model_url = cached_download(hf_hub_url(model_loading_name, filename="pytorch_model.bin"))
|
241 |
+
|
242 |
+
loaded = torch.load(model_url, map_location='cpu')
|
243 |
+
print(loaded)
|
244 |
+
|
245 |
+
model.load_state_dict(loaded)
|
246 |
|
247 |
print(model)
|
248 |
|