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 import math
from dataclasses import dataclass from dataclasses import dataclass
from coqpit import Coqpit
import torch import torch
import torch.nn as nn import torch.nn as nn
from coqpit import Coqpit
from torch.nn import functional as F from torch.nn import functional as F

View File

@ -251,7 +251,7 @@ class ModelManager(object):
else: else:
self._download_zip_file(model_item["github_rls_url"], output_path, self.progress_bar) 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): if isinstance(model_item["hf_url"], list):
self._download_model_files(model_item["hf_url"], output_path, self.progress_bar) self._download_model_files(model_item["hf_url"], output_path, self.progress_bar)
else: else: