Make style

This commit is contained in:
Eren G??lge
2023-06-19 14:22:32 +02:00
parent 37b708dac7
commit f4c88ed677
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -4,10 +4,10 @@ Much of this code is adapted from Andrej Karpathy's NanoGPT
"""
import math
from dataclasses import dataclass
from coqpit import Coqpit
import torch
import torch.nn as nn
from coqpit import Coqpit
from torch.nn import functional as F
+1 -1
View File
@@ -251,7 +251,7 @@ class ModelManager(object):
else:
self._download_zip_file(model_item["github_rls_url"], output_path, self.progress_bar)
def _download_hf_model(self, model_item:Dict, output_path: str):
def _download_hf_model(self, model_item: Dict, output_path: str):
if isinstance(model_item["hf_url"], list):
self._download_model_files(model_item["hf_url"], output_path, self.progress_bar)
else: