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

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

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: